用python语言从包含全班同学信息的.csv格式文件中读入数据,根据用户输入的分组要求进行随机分组,显示分组结果并将其写入.csv格式文件中
时间: 2024-05-01 07:20:17 浏览: 51
用python读取CSV数据
首先,需要安装 `pandas` 和 `numpy` 库,可以通过以下命令进行安装:
```
pip install pandas numpy
```
然后,我们可以编写以下代码实现读取数据并进行随机分组:
```python
import pandas as pd
import numpy as np
# 读取数据
df = pd.read_csv('students.csv')
# 获取学生姓名列表
students = df['姓名'].tolist()
# 获取分组数
n_groups = int(input('请输入分组数:'))
# 随机分组
np.random.shuffle(students)
groups = np.array_split(students, n_groups)
# 显示分组结果
for i, group in enumerate(groups):
print(f'第{i+1}组:{", ".join(group)}')
# 将分组结果写入文件
with open('groups.csv', 'w') as f:
for i, group in enumerate(groups):
f.write(f'第{i+1}组\n')
for student in group:
f.write(f'{student}\n')
```
其中,`students.csv` 是包含全班同学信息的 csv 格式文件,包含一列姓名信息。用户需要输入分组数,程序会随机将学生分为指定的组数,并将结果输出到控制台和 `groups.csv` 文件中。
阅读全文