Python Pandas Excelシート上で同じ値しか入力されていない列を探しだし、その列を削除したい

Question

前提・実現したいこと

Python PandasでExcleの操作を行っています。

同じ値しか入力されていない列を探し、その列を削除したい場合どのような処理が考えられますか？

この状態を

列1	列2	列3	列4	列5
1	1	1	1	1
2	1	2	1	1
3	1	2	3	1
4	1	1	3	4

↓↓↓

列1	列3	列4	列5
1	1	1	1
2	2	1	1
3	2	3	1
4	1	3	4

この状態にする場合、どのような処理をすればいいかお助けください。

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

ソースコード

試したこと

duplicated()を使い考えていたのですが、イメージが湧きません。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

ユニークな要素の数(nunique())が２つ以上ある列を抜き出すと良いかと思います。

import pandas as pd

df = pd.DataFrame([[1,1,1,1,1],
                   [2,1,2,1,1],
                   [3,1,2,3,1],
                   [4,1,1,3,4]],
                  columns=['列1','列2','列3','列4','列5'])

ret = df.loc[:,df.nunique()>1]
#   列1  列3  列4  列5 
#0   1   1   1   1
#1   2   2   1   1
#2   3   2   3   1
#3   4   1   3   4

Comments

[mukku]

ユニークな値から見る発想がありませんでした。ありがとうございます。参考になりました。

Answer 2

特定の列がすべて同じ値を持つかは(df['col'] == df.loc[0,'col']).all()で判定できるので、以下のようなコードで対象列を削除できます。

import pandas as pd

df = pd.DataFrame({'c1':[1,2],'c2':[1,1],'c3':[1,3]})

drops = [c for c in df.columns if (df[c] == df.loc[0,c]).all() ]
df = df.drop(drops, axis=1)
print(df)
"""
   c1  c3
0   1   1
1   2   3
"""

Comments

[mukku]

ありがとうございました。ループして探すケースも作成してみたいと思います。参考になりました。