백준 1620 나는냐 포켓몬 마스터 이다솜 [python]

import sys

N, M = map(int, sys.stdin.readline().split())

poketmon = {}
for i in range(N):
    name = sys.stdin.readline().strip()
    poketmon[name] = i+1


reverse_poketmon = dict(map(reversed, poketmon.items()))


for i in range(M):
    s = sys.stdin.readline().strip()
    if 48 <= ord(s[0]) <= 57:
        print(reverse_poketmon.get(int(s)))
    else:
        print(poketmon.get(s))