dataframe groupby 结果写成dataframe
时间: 2023-08-22 18:06:58 浏览: 45
您好!对于将 DataFrame 的 groupby 结果写回到 DataFrame 中,您可以使用 `transform` 方法。以下是一个示例:
```python
import pandas as pd
# 创建示例 DataFrame
df = pd.DataFrame({'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar'],
'B': ['one', 'one', 'two', 'two', 'three', 'three'],
'C': [1, 2, 3, 4, 5, 6],
'D': [7, 8, 9, 10, 11, 12]})
# 对 A 列进行分组,并计算每组的平均值
df['mean'] = df.groupby('A')['C'].transform('mean')
print(df)
```
输出结果如下:
```
A B C D mean
0 foo one 1 7 3.0
1 bar one 2 8 4.0
2 foo two 3 9 3.0
3 bar two 4 10 4.0
4 foo three 5 11 3.0
5 bar three 6 12 4.0
```
在上述示例中,我们对 DataFrame 的 A 列进行了分组,并使用 `transform` 方法计算了每个分组的 C 列的平均值,并将结果写回到新的列 mean 中。