熊猫groupby键，然后根据组和返回键进行过滤

Question

熊猫执行以下操作的方式是什么？

data.groupby('id').duration.max().index[data.groupby('id').duration.max() > 365]

我不想按ID分组，然后使用组过滤并在条件为真的情况下返回ID。

使用group.filter函数返回原始数据帧。

Answer 1

您可以重写代码-从聚合函数max和index调用Series boolean indexing ：

s = data.groupby('id').duration.max()
idx = s.index[s > 365]
#alternative
#idx = s[s > 365].index

您还可以检查Series过滤值：

print(s[s > 365])

但是，如果要过滤原始DataFrame由max每组值添加GroupBy.transform退货Series具有相同尺寸的原始DataFrame ：

data[data.groupby('id').duration.transform('max') > 365]