PAT甲级1022 Digital Library (python)

本题翻译：
数字图书馆，很多书，需要做个分类，给出第一个数字，共计多少本书，每本书有六条信息，分别是ID、书名、作者、关键词、出版者、出版年份。再给出查找信息，进行书本遍历查找即可。
本题思路：
本题基本是通过二维数组，将数据存储，再遍历查找，最后对查找的ID进行排序即可，注意，需要锁定给出的第几条信息进行遍历，可能查找的信息是第三条，但是第二条也满足，这样的书本不算查找成功的书本。
未完全成功的代码：

#甲级1022
#多少本书
m = int(input())
a = [[] for i in range(3)]
#存放书的信息
for i in range(m):
    for j in range(6):
        a[i].append(str(input()))
#输入查找信息
for i in range(int(input())):
    s =input()
    inp, d = map(str,s.split(":"))
    z = d.replace(d[0],"",1)
    print(s)
    inpint = int(inp)
    cnt = 0
    a1 = []
#每本书进行查找
    for j in range(len(a)):
        if a[j][inpint] == z or (inpint==3 and z in a[j][inpint]):
            cnt+=1
            a1.append(int(a[j][0]))
    if cnt == 0:
        print("Not Found")
    else:
        a1 = sorted(a1)
        for i in a1:
            print(i)

代码总是不完美，只拿到了22分，最后两个没通过，等后续有时间必来查找问题！！！基本思路应该问题不大。