python - 如何根据 "If any observation in a particular column meets a condition, then True?"制作 Pandas 数据框列

Question

我有一个变量，如果其包含组中的任何项目满足条件，我希望将其定义为 True。例如，在下面的 .csv 框架中，与我正在查看的条件相对应的列 D 对于 A==1 的所有行都为 True 因为它在第一行中为 true，False 表示 A==4 因为它在唯一一行中为 False，True 表示A==6 因为对于 A==8 的所有行，两个值为 True 和 False，因为没有是真的。

A,B,C
1,2,True
1,4,False
1,5,False
4,5,False
6,7,True
6,4,False
6,5,True
8,9,False
8,11,False
8,20,False

我尝试使用 .any() 方法，但它始终返回空数据框。

Answer 1

您可以对 A 进行分组，然后使用 transform 保持与原始数据帧相同的形状。应用 lambda 函数，测试 C 列中相应组的任何成员是否为 True。

df['D'] = df.groupby('A').C.transform(lambda group: group.any())

>>> df
   A   B      C      D
0  1   2   True   True
1  1   4  False   True
2  1   5  False   True
3  4   5  False  False
4  6   7   True   True
5  6   4  False   True
6  6   5   True   True
7  8   9  False  False
8  8  11  False  False
9  8  20  False  False