熊猫获取特定其他列值的最大值和最小值

Question

我有以下示例数据集。

 raw_data = {'name': ['Willard Morris', 'Al Jennings', 'Omar Mullins', 'Spencer 
 McDaniel'],
            'age': [20, 19, 22, 21],
            'favorite_color': ['blue', 'blue', 'yellow', "green"],
            'grade': [88, 92, 95, 70]}

 df = pd.DataFrame(raw_data)
 df.head()

我有兴趣找到特定颜色的最高、最低等级，比如“蓝色”，我怎样才能找到它。

我的解决方案如下......有没有更好的方法。我拥有庞大的数据集性能也是选择可用选项时的一个标准。

for i in df.index:
    if df.at[i, 'server_ip'] == 'blue':
        if min_grade > df.at[i,'grade']:
           min_grade = df.at[i,'grade']
        if max_grade < df.at[i,'grade']:
           max_grade = df.at[i,'grade']

Answer 1

我个人会在这里使用 .loc

df.loc[df['favorite_color']=='blue','grade'].max()
df.loc[df['favorite_color']=='blue','grade'].min()

编辑 1

如果不想搜索两次，先保存.loc的结果再进行聚合

data = df.loc[df['favorite_color']=='blue','grade']
min = data.min()
max = data.max()

编辑 2

如果您想搜索每种颜色的最大值和最小值，请使用groupby

grouped = df.groupby('favorite_color')['grade']
max = grouped.max()
min = grouped.min()