当一个单元格值等于特定字符串时,获取列表中的Pandas数据框列名称

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

数据框看起来像这样:

    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]

回答

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

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

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


以上是当一个单元格值等于特定字符串时,获取列表中的Pandas数据框列名称的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>