pandas把一列中相同的元素合并
时间: 2023-11-19 07:12:22 浏览: 259
可以使用 pandas 的 groupby 函数来实现相同元素的合并。
假设我们有一个包含多个重复元素的 DataFrame,如下所示:
```
import pandas as pd
df = pd.DataFrame({'fruit': ['apple', 'banana', 'apple', 'banana', 'apple', 'banana'],
'quantity': [2, 3, 4, 1, 2, 5]})
print(df)
```
输出:
```
fruit quantity
0 apple 2
1 banana 3
2 apple 4
3 banana 1
4 apple 2
5 banana 5
```
现在,我们希望将相同的水果合并,并将其数量相加。
可以使用 groupby 函数来实现:
```
grouped = df.groupby('fruit')
result = grouped.sum()
print(result)
```
输出:
```
quantity
fruit
apple 8
banana 9
```
这里,我们首先使用 groupby 函数按照 'fruit' 列进行分组,然后使用 sum 函数对每个分组中的 'quantity' 列进行求和。最后,我们得到了一个新的 DataFrame,其中每行包含一个唯一的水果和其对应的数量总和。
阅读全文