python-从CSV的文件中提取数据写入EXcel实例演示
python-从CSV的文件中提取数据写入EXcel实例演示
·
代码示例:
需求:
1.先生成一个CSV的文件‘学生信息统计’,共200条数据,对应的字段信息有:学生姓名(中文随机,3位数),年级(1-9随机数),学科(语文、英语、数学、体育,随机取值),成绩(20-100, 取随机数),备注;
2.对‘学生信息统计’统计CSV的文件读取成绩大于60分学生信息数据,写入 Excel 文件‘大于60分’,对应的字段信息有:学生姓名、年级、学科、成绩;
import pandas as pd
import random
import os
# 生成学生信息数据
data = []
subjects = ['语文', '英语', '数学', '体育']
for i in range(200):
student_name = ''.join([chr(random.randint(0x4e00, 0x9fa5)) for _ in range(3)]) # 随机生成中文姓名
grade = random.randint(1, 9)
subject = random.choice(subjects)
score = random.randint(20, 100)
remark = ''
data.append([student_name, grade, subject, score, remark])
# 将数据写入 CSV 文件
df = pd.DataFrame(data, columns=['学生姓名', '年级', '学科', '成绩', '备注'])
df.to_csv('学生信息统计.csv', index=False)
# 从 CSV 文件中筛选成绩大于60分的学生信息
df = pd.read_csv('学生信息统计.csv')
df_gt_60 = df[df['成绩'] > 60][['学生姓名', '年级', '学科', '成绩']]
# 将数据写入 Excel 文件
output_file = '大于60分.xlsx'
if os.path.exists(output_file):
os.remove(output_file) # 如果文件已存在,则先删除
with pd.ExcelWriter(output_file) as writer:
df_gt_60.to_excel(writer, index=False, sheet_name='大于60分')
print(f'成功将成绩大于60分的学生信息写入 {output_file} 文件中。')
更多推荐
已为社区贡献2条内容
所有评论(0)