在列列表上使用 OR 语句过滤 Pandas Dataframe

Question

我想根据列表中的几列中的任何一列是否与测试匹配来过滤数据框。

例如，它可以这样工作：

ddf = pd.DataFrame(np.random.randint(0,100,size=(100, 4)), columns=list('ABCD'))
ddf[(ddf['A']==0)|(ddf['B']==0)|(ddf['C']==0)|(ddf['D']==0)]

...如果有更多的列要处理，则可以构建一个循环。 但我想知道是否有更 Pythonic 的方式来进行，从结果开始

ddf[list('ABCD')]==0

它给出了 4 列布尔值，我想按行应用or操作。

Answer 1

如果它是相同的测试，例如它是否为零，那么您可以any()使用any() ：

np.random.seed(999)
ddf = pd.DataFrame(np.random.randint(0,100,size=(100, 4)), columns=list('ABCD'))

ddf[(ddf[['A','B','C','D']]==0).any(axis=1)]

     A   B   C   D
71  52  13   0  50
93  51   0  60  71