Conditional filter multiple columns python

Question

I have 5 columns, each of them has 0's and 1's in each row. I need to filter all those with '1' at once.

I tried this but results in error:

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

Answer 1

I believe you need any if want filter at least one 1 per rows of filtered columns:

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

Or all if want filter all 1 per rows of filtered columns:

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