python - groupby 操作后过滤掉数据帧

Question

我正在尝试对以下数据框进行分组，并在分组后进行过滤。

如何检查 col2 对应的值对于 col1 组是否为 True

输入数据框看起来像

col1    col2    col3
1   True    p1
1   True    p2
1   True    p3
1   True    p4
2   False   p1
2   True    p2
2   True    p3
2   True    p4
3   False   p1
3   False   p2
3   False   p3
3   False   p4
4   True    p1
4   True    p2
4   True    p3
4   True    p4

我尝试了groupby和过滤器

filtered_df = df.groupby(['col1']).filter(lambda final_filter : final_filter['col2'] is True)

最终输出如下

输出:

col1    col2    col3
1   True    p1
1   True    p2
1   True    p3
1   True    p4
4   True    p1
4   True    p2
4   True    p3
4   True    p4

Answer 1

您可以在过滤器中对“col2”使用all:

print (df.groupby('col1').filter(lambda x: x.col2.all()))
    col1  col2 col3
0      1  True   p1
1      1  True   p2
2      1  True   p3
3      1  True   p4
12     4  True   p1
13     4  True   p2
14     4  True   p3
15     4  True   p4