根據條件從 pandas dataframe 中提取行

Question

我有 pandas dataframe “數據”，並且只想保留每個類別“類”的“numb_people”總和至少為 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])