python - 检查某个字符串的 Dataframe 并返回在其中找到该字符串的列的列标题

Question

我有一个看起来像这样的数据框:

现在我只想将具有字符串“worked”的列的标题返回到列表中。

因此在这种情况下，列表仅包含 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']

因此，解决方案构建了一个 bool 系列，其中列包含术语“Worked”。然后，我们只获取与真实标签相关的系列部分，通过调用 index 选择标签并将该对象作为列表返回