Python Pandas で特定の値を含んだ列の名称を抽出したい

Question

前提・実現したいこと

Pandas Dataframeにおいて、
特定の値を含んだ列の名称の抽出を行いたいです。

###データフレーム
df1

|| A | B | C | D |
1|false|false|false|false|
2|false|false|false|false|
3|false|true |false|false|
4|false|false|false|false|
5|false|false|false|true |

上記のデータフレームにおいて、
B列とD列に"true"が含まれるので、
この場合の出力を[ B , D ]としたいです。

Answer 1

条件を満たすDataFrameを作成し、それからcolumnsを取得しリスト化します。

import pandas as pd

df = pd.DataFrame({'A':[False,False], 'B':[True, False]})
ret = list(df.loc[:,df.any()].columns)
print(ret) # ['B']

Comments

[yuyuy1]

any(),not any()で判別して吐き出せば良かったのですね。
ご回答ありがとうございました。