python - Pandas 按多列的单一条件过滤数据帧

Question

我有一个包含 30 列的数据框，每个单元格包含 1 或 0。

我想过滤数据帧，以查找 4 个特定列中的任意 1 列 == 1。

现在，它必须是链式掩码，例如:

df[(df['col1'] == 1) | (df['col2'] == 1) | (df['col3'] == 1) | (df['col4'] == 1)]

是否有更简单且可扩展的方法来做到这一点？比如制作列名列表，并使用切片表示法一次性检查所有 bool 条件？:

col_list = ['col1', 'col2', 'col3', 'col4']

df[df[col_list] == 1]

Answer 1

试试这个:

col_df=list(df) # get all the column names 
filtered_df=df[df[col_df].any(1)]