基于Pandas按Electronics分组求和Price并筛选总价超1200的实现

输入数据 #

Electronics            Price

IPhone 11              500 

Iphone 11 pro          800

Iphone 11 pro max      1200

IPhone 11              550  

Iphone 11 pro          850

Samsung S 20           750

Iphone 11 pro max      1250

Samsung S 20 pro       1400

Samsung S 22           1000

Samsung S 20           800

Samsung S 20 pro       1350

代码实现 #

import pandas as pd

# 构造DataFrame（如果数据来自本地文件，替换成pd.read_csv('你的文件路径.csv')即可）
df = pd.DataFrame({
    'Electronics': [
        'IPhone 11', 'Iphone 11 pro', 'Iphone 11 pro max',
        'IPhone 11', 'Iphone 11 pro', 'Samsung S 20',
        'Iphone 11 pro max', 'Samsung S 20 pro', 'Samsung S 22',
        'Samsung S 20', 'Samsung S 20 pro'
    ],
    'Price': [500, 800, 1200, 550, 850, 750, 1250, 1400, 1000, 800, 1350]
})

# 按电子产品分组求和，再筛选总价超过1200的条目
total_price = df.groupby('Electronics')['Price'].sum()
filtered_result = total_price[total_price > 1200]

# 输出最终结果
print(filtered_result)

输出结果 #

Iphone 11 pro        1650
Iphone 11 pro max    2450
Samsung S 20         1550
Samsung S 20 pro     2750

注：原预期输出中Iphone 11 pro max的总价标注为2540，实际计算应为1200+1250=2450，属于笔误。

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