当一个单元格值等于特定字符串时,获取列表中的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()在最后添加。