pat乙级1031 查验身份证 (15 分)较精简代码 python

很简单的题 可以用列表和异常处理来简化
见代码
sys.stdin.readline()和input()作用一样 但是性能更优

import sys
#权重与对应关系 
weight = [7, 9, 10, 5, 8,4,2,1,6,3,7,9,10,5,8,4,2]
M=["1","0","X","9","8","7","6","5","4","3","2"]

many=int(sys.stdin.readline())
result=[]
for i in range(0,many):
    try:
        input_id=sys.stdin.readline()
        #题干指出了输入的身份证号一定是18位
        tem_id=[list(map(int,input_id[0:17])),input_id[17]]
        last=0
        for index in range(0,17):
        	#直接用index去权重列表中获取权重
            last=last+tem_id[0][index]*weight[index]
        last=last%11
        #M中的index 就对应M[index] 所以直接检验相等
        if M[last]==tem_id[1]:
            continue
        else:
        	#手动抛出异常 表示这个身份证不正确
            raise ValueError
    except(ValueError):
    	#异常处理 以下几种情况都会进入到这一步
    	#前17位有除了数字意外的字符 会导致python无法对其进行int化
    	#最后位数对应不一样 手动抛出ValueError异常进入异常处理
        result.append(input_id.rstrip())
if result:#列表长度不为零时
    for x in result:
        print(x)
else:
    print("All passed")

有更优化的代码的话 欢迎大佬指教