代码示例:
需求:
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} 文件中。')

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐