学生成绩录入及查询 python

7-5 学生成绩录入及查询 (10 分)

学生成绩表中，一名学生的信息包含如下信息： 学号(11位)、姓名、数学成绩、英语成绩、程序设计成绩、物理成绩。

本题要求编写程序，录入N条学生的信息，并且按查询要求显示学生成绩等信息。

输入格式:
输入在第一行给出正整数N(N<=10)；随后N行，每行按格式：

学号 姓名 数学成绩 英语成绩 程序设计成绩 物理成绩
给出一名学生的信息。 其中学号是11位数字；姓名是不超过10个字符、不包含空格的非空字符串；成绩是<=100的非负整数；

在学生信息输入完成后，给出要查询的姓名。

输出格式:
对查询的姓名的每一名学生，在一行中按照如下格式输出：

学号 姓名 数学成绩 英语成绩 程序设计成绩 物理成绩
如果有相同姓名的同学，输出全部同学的信息，每名同学一行； 如果要查询的学生不存在，则输出：

Not Found！
输入样例:
在这里给出一组输入。例如：

4
20201003012 LiWen 68 90 78 86
20201003012 HeHua 91 88 80 82
20201002121 WangFang 78 78 82 90
20201002112 ZhengYi 56 83 77 80
ZhengYi

输出样例:
在这里给出相应的输出。例如：

20201002112 ZhengYi 56 83 77 80

代码如下：

n = int(input())
flag=0
l=[list(input().split()) for i in range(n)]
s=input()
for i in l:
    if i[1]==s:
        print(' '.join(i))
        flag=1
if flag==0:
    print('Not Found!')