cpy3, pandas: 欠損値がある column の名前をリストにするのをワンライナーで

Question

以下のループをなくしてワンライナーで記述する方法などありますか？もっと簡潔にしたいです

import pandas as pd
df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, float('nan'), 5], 'c': [6, 7, 8]}, index=[1,2,3])

ans = []
for c,t in zip(df, df.isnull().any()):
    if not t: continue
    ans.append(c)

assert ans == ['b']

環境

yosuke@localhost:~$ python --version && python -m pip list | grep pandas
Python 3.11.2
pandas             2.2.3
pandas_ta          0.3.14b0

Answer 1

Pythonはあまり使わないのでよく分かりませんが，

ans = list(df.columns[df.isnull().any()])

とかでしょうか?

Comments

[yosukesan]

ありがとうございました。実行速度もおそらく pandas 側なので問題なさそうです。FA とさせていただきます