python - 列出符合条件的列名

Question

我在每个单元格中都有一个带有 FALSE 和 TRUE 的数据框...我想制作一个列表，其中包含至少一个 TRUE 的列的名称。

 col1 col2 col3 col4  
FALSE FALSE TRUE FALSE  
FALSE TRUE FALSE FALSE  
FALSE TRUE FALSE TRUE

我想要这个输出：

[col2, col3, col4]

我该怎么做？

标签： pythonlistdataframe

Answer 1

这个简单的代码做到了：

colList = []
for col in df.columns:
    if True in df[col].unique():
        colList.append(col)

这为您提供了colList其中包含的列的名称True。（假设您的数据框名为df）