使用Python Pandas分组计算酒店收入的min、median、max并输出Abu结果

原始数据 #

需求 #

使用Pandas按Hotel字段分组，计算每家酒店Earning的最小值（min）、中位数（median）和最大值（max），并输出酒店名称为“Abu”的聚合结果。

解决方案代码 #

import pandas as pd

# 构造数据
data = {
    'Hotel': ['Abu', 'Zain', 'Show', 'Zint', 'Abu', 'Zain', 'Abu', 'Abu', 'Abu', 'Abu', 'Show', 'Zint'],
    'Earning': [1000, 400, 500, 300, 500, 700, 500, 500, 800, 1600, 1300, 600]
}
df = pd.DataFrame(data)

# 分组聚合计算指定统计值
agg_result = df.groupby('Hotel')['Earning'].agg(['min', 'median', 'max'])

# 提取Abu的结果并转为列表
abu_result = agg_result.loc['Abu'].tolist()
print(abu_result)

运行结果 #

[500.0, 650.0, 1600.0]

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