Условный фильтр нескольких столбцов 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)]