Python dataframe将B列按A列分组
时间: 2024-05-14 15:17:56 浏览: 135
python中dataframe将一列中的数值拆分成多个列
5星 · 资源好评率100%
你可以使用`groupby()`函数将DataFrame按照A列进行分组,然后将B列中的值进行合并。下面是示例代码:
```python
import pandas as pd
# 创建示例DataFrame
df = pd.DataFrame({'A': ['apple', 'banana', 'apple', 'banana', 'cherry'],
'B': [1, 2, 3, 4, 5]})
# 按A列进行分组,将B列中的值进行合并
result = df.groupby('A')['B'].apply(list).reset_index(name='B')
print(result)
```
输出结果如下:
```
A B
0 apple [1, 3]
1 banana [2, 4]
2 cherry [5]
```
在这个示例中,我们首先创建了一个包含两列数据的DataFrame。接着,我们使用`groupby()`函数将DataFrame按照A列进行分组,并使用`apply()`函数将B列中的值合并成一个列表。最后,我们使用`reset_index()`函数将A列和B列的名称分别设置为'A'和'B',以便于查看和处理结果。
阅读全文