Проверьте Dataframe для определенной строки и верните заголовки столбцов столбцов, в которых находится строка

Question

У меня есть датафрейм, который выглядит примерно так:

Теперь я просто хочу вернуть заголовки столбцов, в которых есть строка «сработало», в список.

Так что в этом случае список включает только lst = ["OBE"]

Answer 1

Вы можете получить это так:

import pandas as pd
import numpy as np

df = pd.DataFrame({'OBE': ['Worked', 'Worked', np.nan, 'Uploaded'],
                   'TDG': ['Uploaded']*4,
                   'TMA':[np.nan]*4, 'TMCZ': ['Uploaded']*4})
columns_with_worked = (df == 'Worked').any(axis=0)
columns_with_worked[columns_with_worked].index.tolist()

['OBE']

Таким образом, решения составляют логическую серию, в столбцах которой содержится термин «Обработано». Затем мы получаем только часть серии, относящуюся к истинной метке, выбираем метки, вызывая index и возвращаем этот объект в виде списка