Python Pandas DataFrame按Fruit列相同值对Quantity列求和实现方法

你可以直接使用Pandas内置的groupby分组聚合功能实现需求，具体操作如下：

核心代码 #

假设你的原始DataFrame命名为df，执行以下代码即可得到目标结果：

# 按Fruit列分组，对Quantity列求和，且保留Fruit为普通列而非索引
res = df.groupby('Fruit', as_index=False)['Quantity'].sum()

参数说明 #

• groupby('Fruit')：指定按照Fruit列的唯一取值对数据进行分组
• as_index=False：设置后分组键Fruit不会被设为行索引，保持普通列结构，匹配你需要的输出格式
• ['Quantity'].sum()：仅对分组后的Quantity列执行求和聚合计算

完整可运行示例 #

你可以直接运行以下代码验证效果：

import pandas as pd

# 构造你给出的原始数据集
df = pd.DataFrame({
    'Fruit': ['orange', 'grape', 'apple', 'grape', 'orange'],
    'Quantity': [4, 2, 3, 2, 1]
})

# 执行分组求和
res = df.groupby('Fruit', as_index=False)['Quantity'].sum()

# 打印结果
print(res)

注：默认groupby会按分组列的字典序排序输出，如果你需要和示例中完全一致的输出顺序，可以额外添加自定义排序逻辑即可。

内容的提问来源于stack exchange，提问作者jumpman8947