python - 当一个单元格值等于特定字符串时，获取列表中的 pandas 数据框列名

Question

我有一个大约 20-30 列的熊猫数据框。当一个单元格值等于字符串“失败”时，我想获取列表中的列名。

数据框如下所示：

    Column_1 Column_2 ... Column_30
0      Pass   Pass          Fail
1      Pass   Pass          Pass
2      Fail   Pass          Pass
3      Pass   Pass          Pass
4      Pass   Pass          Pass
..

预期的结果看起来像这样[Column_1, Column 30]

标签： pythonpandaslistdataframe

Answer 1

我认为这应该可以解决问题

df.columns[(df=='Fail').any(axis=0)]

这将为您提供在其行中至少包含一个“失败”的每一列的列名称作为 pandas.Index。如果必须是列表，只需.tolist()在末尾添加即可。