基于条件从pandas数据帧中提取行

Question

我有pandas dataframe“data”，只想保留求和的行 “麻木的人”每个类别的“类”至少是2

但是，这会引发索引错误（索引不再匹配）：

data = data[data.groupby('class').sum()['numb_people'] > 2]

我怎样才能以同样简单的方式做到这一点

Answer 1

正如@rafaelc在评论中所说：

idx = data.groupby('class').numb_people.transform('sum') > 2
print(data[idx])