分组后的数据如何输出为csv
时间: 2024-04-12 10:26:55 浏览: 67
IMDB电影数据集 CSV 2K 记录
5星 · 资源好评率100%
在将分组后的数据输出为CSV文件时,可以使用`to_csv`方法。下面是一个示例代码:
```python
import pandas as pd
# 创建一个示例数据集
data = {'A': ['A', 'A', 'B', 'B', 'B'],
'B': ['X', 'Y', 'X', 'Y', 'Z'],
'C': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)
# 按照字段 A 和 B 进行分组,并计算字段 C 的和
grouped_data = df.groupby(['A', 'B'])['C'].sum()
# 将分组后的数据输出为CSV文件
grouped_data.to_csv('grouped_data.csv')
```
在这个示例中,我们创建了一个包含字段 A、B 和 C 的数据集。然后,我们使用`groupby`函数按照字段 A 和 B 进行分组,并计算字段 C 的和。接下来,我们将分组后的数据保存为名为`grouped_data.csv`的CSV文件。你可以在代码的最后指定你想要保存的文件名和路径。
阅读全文