【问题描述】
现有学生成绩单文件class.csv,存有包括1班和2班的学生学号,姓名,班级和成绩,在控制台输入班级,根据班级编程实现成绩单筛选并将筛选的成绩信息存入同目录下的文件class_01.csv里
class.csv
【输入形式】文件class.csv
【输出形式】根据控制台输入,进行班级划分,结果存入class_01.csv
【样例输入】1班
【样例输出】
class_01.png
【样例输入】2班
【样例输出】
【样例说明】
【评分标准】
import csv
b=input()
with open('class.csv', 'r') as f:
reader = csv.reader(f)
with open('class_01.csv', 'a') as ff:
a=0
for row in reader:
if a == 0:
a += 1
writer = csv.writer(ff)
writer.writerow(row)
continue
else:
if row[2]==b:
writer.writerow(row)
else:
continue