根据条件删除整个组的 pandas 行

Question

import seaborn
df = seaborn.load_dataset('flights')

我想放弃每年平均乘客人数少于 200 人的年份。我试过这个

df[df.groupby(['year'])['passengers'].mean() > 200] 

但得到这个错误：

*** pandas.core.indexing.IndexingError: Unalignable boolean Series provided as indexer (index of the boolean Series and of the indexed object do not match).

在正确答案中，数据框应删除这些年份的行： 1949、1950、1951、1952

Answer 1

我认为，你需要：

• 按年分组，
• 过滤组，检查 passengers 在 当前组 > 300。

所以代码应该是：

df.groupby(['year']).filter(lambda x: x.passengers.mean() > 300)