python的pandas分组合并
时间: 2023-05-18 18:04:07 浏览: 143
Numpy&pandas(四)–分组计算
可以使用 pandas 的 groupby 和 merge 方法来实现分组合并。首先使用 groupby 方法将数据按照指定的列进行分组,然后使用 merge 方法将分组后的数据进行合并。以下是示例代码:
```python
import pandas as pd
# 创建示例数据
df1 = pd.DataFrame({'key': ['A', 'B', 'C', 'D'], 'value': [1, 2, 3, 4]})
df2 = pd.DataFrame({'key': ['B', 'D', 'E', 'F'], 'value': [5, 6, 7, 8]})
# 按照 key 列进行分组
grouped = df1.groupby('key')
# 对每个分组进行合并
result = pd.concat([grouped.get_group(x).merge(df2, on='key') for x in grouped.groups])
print(result)
```
输出结果为:
```
key value_x value_y
0 B 2 5
1 D 4 6
```
其中,value_x 和 value_y 分别表示两个数据集中的 value 列。
阅读全文