複数の列の全ての値からユニークな値を取り出す

Question

pandasのDataFrame形式のデータの複数の列から、ユニークな値を取り出して表示したいのですが、
2列からユニークな値を取り出すとエラーが発生してしまいます。

import pandas as pd

# 6行×2列のデータフレームを作る
df = pd.DataFrame({"A":[1,1,2,3,1,1],"B":[5,4,3,4,6,7]})

# A列と、B列の全ての値からユニークな値を取り出す
unique_list = pd.unique(df['A', 'B']).tolist()
unique_list

ここで出てしまうエラーメッセージは以下の通りです。

KeyError: ('A', 'B')

コードの間違いを指摘いただけますでしょうか。

Comments

[hentaiman]

Aだけでユニーク＆Bだけでユニークなのか、AとB全てまとめてユニークなのかどちら？

[hentaiman]

結果として欲しい形のlistを追記してくれるのでもOK

[color_8]

hentaimanさん
有難う御座います。

A列とB列をすべてまとめてからのユニークを意図してます。
たとえば、A列にも、B列にも同じ値が入ってきてしまうことも考慮しています。

イメージ的には、A列とB列をマージさせ、
そのうえでユニークにし、昇順で値を表示させる操作です。

Answer 1

import pandas as pd
from itertools import chain
df = pd.DataFrame({"A":[1,1,2,3,1,1],"B":[5,4,3,4,6,7]})
uniql = list(set(chain.from_iterable(df.values)))

Comments

[color_8]

ご回答有難う御座いました！解決いたしました。

Answer 2

DataFrame から列を取り出すときは df[['A', 'B']] とするようですが、
その後の unique() が実行できないようです

こんなコードはいかがでしょうか:

import pandas as pd

# 6行×2列のデータフレームを作る
df = pd.DataFrame({"A":[1,1,2,3,1,1],"B":[5,4,3,4,6,7]})
# A列と、B列の全ての値からユニークな値を取り出す
unique_list = [df[column_name].unique().tolist() for column_name in ['A', 'B']]
print(unique_list)

実行結果:

$ pipenv run python test.py
[[1, 2, 3], [5, 4, 3, 6, 7]]

Comments

[color_8]

y_shinodaさん
有難う御座います。

頂いた結果だと、3が2つ入ってきてしまいます。

イメージ的には、A列とB列をマージさせ、
そのうえでユニークにし、昇順で値を表示させるような感じなのですがご教示いただけますでしょうか。