列の要素が全て0または一つでも0があった場合にその列を削除する

Question

皆様

下記を実現する場合どのようなコードが考えられるでしょうか。
①全て0の列を削除
②1つでも0が含まれていたら削除

↓画像にてイメージを共有します↓

df_Sales

①全て0の列を削除(df_Sales_1)

②1つでも0が含まれていたら削除(df_Sales_2)

どうぞよろしくお願いいたします。

Comments

[hentaiman]

既に実現できるコードがあってより良くする方法を知りたいという事ですか？そういう事なら元のコードを載せてくれますか？

Answer 1

考え方
1.DataFrameに対してゼロとの比較をする
2.比較結果を縦方向に合計する
3.合計が行数と同じならオールゼロと判定する
4.上記に該当しないがゼロより大きければ、ゼロが含まれると判定する

コードイメージ

tmp=df==0
col=df.columns.valus[tmp.sum(axis=0)==len(df)] #オールゼロのカラム取得

Comments

[D7U]

不躾な質問に対してご丁寧なご回答誠にありがとうございます。
1行でここまでスマートに表現できるとは驚きでした。

Answer 2

①は(df[列] == 0).all()、②は(df[列] == 0).any()が使えます。

import pandas as pd

df = pd.DataFrame({'A':[1,2,3],'B':[0,0,0],'D':[7,8,0]})

df1 = df.copy()
for c in df1.columns:
    if (df1[c] == 0).all():
        df1.drop(c, axis=1, inplace=True)
print(df1)

df2 = df.copy()
for c in df2.columns:
    if (df2[c] == 0).any():
        df2.drop(c, axis=1, inplace=True)
print(df2)

Comments

[D7U]

不躾な質問なのにご丁寧な回答誠にありがとうございます！
大変分かりやすいです。

Answer 3

順番に見ていって、０以外が見つかったらなにもしない。
最後まで行ってしまったら削除。

順番に見ていって、０が見つかったら削除
最後まで行ってしまったらなにもしない

この２通りでどうぞ。

Comments

[Zuishin]

それだと全列消えると思います。