PAT (Basic Level) Practice Python解法 1076 Wifi密码(字典与list的替换)
n = int(input())
wifi = []
pw = []
for i in range(n):
wifi.append(input().split(' '))
for hang in wifi:
for pair in hang:
if pair[-1]=='T':
pw.append(pair[0])
break
check = {'A':1,'B':2,'C':3,'D':4}
for item in pw:
print(check[item],end='')
想用
list
的一一替换可以参考这个写法 click me