python现有学生成绩单文件class.csv,存有包括1班和2班的学生学号,姓名,班级和成绩,在控制台输入班级,根据班级编程实现成绩单筛选并将筛选的成绩信息存入同目录下的文件class_01.csv里 class.csv
时间: 2024-02-24 08:54:35 浏览: 78
以下是实现该功能的Python代码:
```python
import csv
# 读取原始成绩单文件
with open("class.csv", "r", encoding="utf-8") as f:
reader = csv.reader(f)
# 创建空列表存储符合条件的学生信息
selected_students = []
# 遍历每一行记录
for row in reader:
# 当班级与输入匹配时,将该行记录加入选中列表
if row[2] == input("请输入班级:"):
selected_students.append(row)
# 将选中的学生信息写入新文件
with open("class_01.csv", "w", encoding="utf-8", newline="") as f:
writer = csv.writer(f)
# 写入表头
writer.writerow(["学号", "姓名", "班级", "成绩"])
# 遍历选中的学生信息列表,逐一写入新文件
for row in selected_students:
writer.writerow(row)
```
以上代码使用了Python自带的csv模块来读写csv文件。首先,我们读取原始成绩单文件并遍历其中的每一行记录,将班级与用户输入进行匹配,符合条件的学生信息被加入一个新的列表`selected_students`。然后,我们打开一个新的文件`class_01.csv`,使用csv.writer来逐行写入选中的学生信息,同时加入表头。需要注意的是,写入csv文件时,需要指定`encoding="utf-8"`来避免中文乱码,同时指定`newline=""`来避免写入空行。
阅读全文