Seabornの散布図で各プロットに色を付けたい。ValueError: The palette list has the wrong number of colors.

Question

Seabornの散布図でプロットを色分けしたいと思い次のコードを書きました。

sns.scatterplot(x=data['block'], y=data[col], hue=data2[col], s=100)
plt.savefig(graph_path + col+r".svg")

おおむね狙い通りの図が描けたのですが、色味が気に入らなかったので次のように書き換えました。

colp = sns.blend_palette(['grey', 'yellow', 'red'], 3)
sns.scatterplot(x=data['block'], y=data[col], hue=data2[col], palette=colp, s=100)
plt.savefig(graph_path + col+r".svg")
・
・
・
ValueError: The palette list has the wrong number of colors.

パレットの色数が気に入らないといわれたので、3を1~4まで、および"なし"で変化させましたが同じ結果でした。
ちなみに data2[col] がとりうるあたいは、0,1,2のどれかです。
どうしたら良いでしょうか？

どうやら失敗しているパターンでは、print(set(data2[col]))の結果が{0,1}のようです。
0,1,2のどれが含まれているか含まれていないかにかかわらず、
0->grey
1->yellow
2->red
となるようにしたいのですが。

Comments

[jbpb0]

colp = ['grey', 'yellow', 'red']

で、どうでしょうか？

[i113]

どうも有り難うございます。
やってみましたが、同じ結果になります。

[8524ba23]

> ちなみに data2[col] がとりうるあたいは、0,1,2のどれかです。
hueだけはdataではなくdata2のほうを参照しているのですね。
これが確実か確認したいので、「print(set(data2[col]))」した結果（「{0,1,2}」になっているか？）を質問本文に追記ください。

Answer 1

どうやら失敗しているパターンでは、print(set(data2[col]))の結果が{0,1}のようです。

コード全体の動きが不明ですが、col毎にカテゴリ数が異なるのであれば、以下のような感じでそれに応じたパレットをつど作成すればよいかと思います。

base_colors = ['grey', 'yellow', 'red'] # 最大のカテゴリ数分だけ用意しておく
：
N = len(set(data2[col]))
colors = base_colors[:N]
colp = sns.blend_palette(colors, N)

追記

↑だと[1,2]のようにカテゴリが0から連続していないと、同じ値なのにcol毎に色が異なることになります。
よって、以下のような値と色の辞書を指定するとよいと思います。

colp = {0:'grey', 1:'yellow', 2:'red'}

Comments

[jbpb0]

それだと、data2が0, 2だけや1, 2だけの場合に、

> 0,1,2のどれが含まれているか含まれていないかにかかわらず、
0->grey
1->yellow
2->red
となるようにしたい

とはならず、数字とは無関係に、一つ目がgray、二つ目がyellowになりませんか？

[i113]

どうも有り難うございます。参考にさせていただき、次のようにしたら解決しました。
colors = ['grey', 'yellow', 'red']
colp = []
for i in set(data2[col]):
colp.append(colors[i])
colp = sns.blend_palette(colp,len(colp))

[8524ba23]

> jbpb0さん
ですね。。
というか単純に「colp = {0:'grey', 1:'yellow', 2:'red'}」という辞書で試してみたら動作しました。
このほうがよさそうです。

[jbpb0]

質問者さん

> 次のようにしたら解決しました。

下記の行、要りますか？

colp = sns.blend_palette(colp,len(colp))

当方のmacのpythonで、下記のようにして適当にデータを作って確認したところ、「sns.blend_palette()」の行が有っても無くても表示されるグラフは同じでした

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

X = np.random.normal(0, 10, 100)
Y = [np.random.normal(0, 5)+i*1.3 for i in X]
M = np.random.randint(0, 2, 100)+1

colors = ['grey', 'yellow', 'red']
colp = []
for i in set(M):
colp.append(colors[i])

#colp = sns.blend_palette(colp, len(colp))
sns.scatterplot(x=X, y=Y, hue=M, palette=colp, s=100)
plt.show()