python - Pandas :按满足条件的列分组

Question

我有一个包含三列的数据集:评级、品种和狗。

import pandas as pd
dogs = {'breed': ['Chihuahua', 'Chihuahua', 'Dalmatian', 'Sphynx'],
        'dog': [True, True, True, False],
        'rating': [8.0, 9.0, 10.0, 7.0]}

df = pd.DataFrame(data=dogs)

我想计算狗为 True 的每个品种的平均 评级。这是预期的:

  breed     rating
0 Chihuahua 8.5   
1 Dalmatian 10.0  

这是我的尝试:

df.groupby('breed')['rating'].mean().where(dog == True)

这是我得到的错误:

NameError: name 'dog' is not defined

但是当我尝试添加 where 条件时，我只会收到错误。谁能建议解决方案？时间差

Answer 1

一旦您分组并选择一列，您的 dog 列就不再存在于您选择的上下文中(即使存在，您也无法正确访问它)。

首先过滤数据框，然后使用 groupby 和 mean

df[df.dog].groupby('breed')['rating'].mean().reset_index()

       breed  rating
0  Chihuahua     8.5
1  Dalmatian    10.0