python-从CSV的文件中提取数据写入EXcel实例演示

美少女战士1@

已于 2024-02-23 17:55:46 修改

阅读量314

点赞数 3

分类专栏：学习笔记文章标签： python excel 开发语言

于 2024-02-23 17:54:38 首次发布

本文链接：https://blog.csdn.net/weixin_48134870/article/details/136261477

版权

学习笔记专栏收录该内容

50 篇文章 1 订阅

订阅专栏

本文介绍了一个使用Pythonpandas库生成包含随机学生信息的CSV文件，然后筛选出成绩大于60分的学生数据，并将这些信息写入Excel文件的过程。

摘要由CSDN通过智能技术生成

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