条件过滤多列 python

Question

我有 5 列，每列的每一行都有 0 和 1。 我需要一次过滤所有带有“1”的人。

我尝试了这个但导致错误：

df_2 = df_1[df_1.columns[0:5] == 1]
ValueError: Item wrong length 2 instead of 111249

Answer 1

我相信你需要any如果想要过滤每行过滤列至少一个1 ：

df_2 = df_1[(df_1.columns[0:5] == 1).any(axis=1)]

或者如果想要过滤all过滤列的所有1行：

df_2 = df_1[(df_1.columns[0:5] == 1).all(axis=1)]