遍历数据框中的多列

Question

我有一个来自具有61列和100万行的CSV文件的数据框。 这些列中的25个列(Flag_1, Flag_2, ..., Flag_25)具有True/False作为数据帧每一行的值。

我想做的是遍历每一列，以确定这些列中的整个行是否存在True，我只需要最少一个true。 如果存在True，则有一个新列， Flag_All将对该行具有True值，如果没有，则为False。

我可以像这样将for循环用于单列，

for index, x in data2['FLAG_1'].iteritems() :
    data2['FLAG_ALL'] =  data2['FLAG_1'] == True

但无法找出多个列。

Answer 1

给定一个示例数据框：

df = pd.DataFrame({
    'flag_1': [False, False, True],
    'flag_2': [False, False, False],
    'flag_3': [True, False, False]})

您可以使用df.filter来获取适当的列（以标志，下划线然后是数字...开头的列），然后在行轴上应用any()以获得整个布尔列：

df['flag_all'] = df.filter(regex='^flag_\d+$').any(axis=1)

这给你：

  flag_1 flag_2 flag_3 flag_all
0  False  False   True     True
1  False  False  False    False
2   True  False  False     True

Answer 2

请试试：

data2['FLAG_ALL'] = data2.any(axis=1,bool_only=True).values