PTA 1044 火星数字(Python3)

火星人是以 13 进制计数的：

• 地球人的 0 被火星人称为 tret。
• 地球人数字 1 到 12 的火星文分别为：jan, feb, mar, apr, may, jun, jly, aug, sep, oct, nov, dec。
• 火星人将进位以后的 12 个高位数字分别称为：tam, hel, maa, huh, tou, kes, hei, elo, syy, lok, mer, jou。

例如地球人的数字 29 翻译成火星文就是 hel mar；而火星文 elo nov 对应地球数字 115。为了方便交流，请你编写程序实现地球和火星数字之间的互译。

输入格式：

输入第一行给出一个正整数 N（<100），随后 N 行，每行给出一个 [0, 169) 区间内的数字 —— 或者是地球文，或者是火星文。

输出格式：

对应输入的每一行，在一行中输出翻译后的另一种语言的数字。

输入样例：

4
29
5
elo nov
tam

输出样例：

hel mar
may
115
13

提交结果：

代码： 

x = ['tret', 'jan', 'feb', 'mar', 'apr', 'may', 'jun', 'jly', 'aug', 'sep', 'oct', 'nov', 'dec']
y = ['', 'tam', 'hel', 'maa', 'huh', 'tou', 'kes', 'hei', 'elo', 'syy', 'lok', 'mer', 'jou']


def earth(num):
    a1 = num // 13
    a2 = num % 13
    if a1 == 0:
        print(x[a2])
    else:
        if a2 == 0:
            print(y[a1])
        else:
            print(y[a1], x[a2])


def huoxin(str1):
    a = str1.split()
    if len(a) == 1:
        if a[0] in x:
            num = x.index(a[0])
        else:
            num = y.index(a[0]) * 13
        print(num)
    else:
        num = y.index(a[0]) * 13 + x.index(a[1])
        print(num)


num = int(input())
for i in range(num):
    data = input()
    if data.isdigit() == True:
        earth(int(data))
    else:
        huoxin(data)