1076. Wifi密码 (15)——Python

def judge(string):
	#判断那个正确
	for my_string in string:
		if my_string[-1] == 'T':
			return my_string[0]

if __name__ == '__main__':
	dir_scores = {'A':'1', 'B':'2', 'C':'3', 'D':'4'}
	num = int(input())
	list_WIFI_password = []
	for i in range(num):
		flag = ''
		string = input().split()
		flag = judge(string)
		list_WIFI_password.append(dir_scores[flag])
	print(''.join(list_WIFI_password))