Python如何从pandas dataframe分组均值表中查询极值对应的区域

实现方法 #

你可以直接使用pandas内置的idxmax()和idxmin()方法完成需求，这两个方法会返回指定列极值对应的行索引，刚好你生成的均值表索引就是区域名称，完全匹配使用场景。

示例代码：

# 获取SAFETY数值最大的区域
max_safety_area = mean['SAFETY'].idxmax()
print(f"SAFETY数值最大的区域：{max_safety_area}")

# 获取TRUST数值最低的区域
min_trust_area = mean['TRUST'].idxmin()
print(f"TRUST数值最低的区域：{min_trust_area}")

运行结果（匹配你提供的示例数据）：

SAFETY数值最大的区域：NORTH
TRUST数值最低的区域：area_4

如果需要同时获取极值的数值，可以搭配max()/min()方法使用：

# 同时获取区域和对应数值
max_safety_val = mean['SAFETY'].max()
print(f"SAFETY数值最大的区域是{max_safety_area}，对应均值为{max_safety_val:.2f}")

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