Seabornの散布図の凡例にラベルを振りなおしたい

Question

次のようなコードを書いてFig.1を得ました。

import seaborn as sns
import matplotlib.pylab as plt

x=["a","b","c","d","e","f"]
y=[1,2,3,4,5,6]
h=[0,0,-1,-1,+1,+1]
s=[10,20,40,10,20,40]
sns.scatterplot(x=x, y=y, hue=h, size=s, legend="full")

Fig. 1

概ね望み通りなのですが凡例のラベルが気に入らないので次のようにしました。

import seaborn as sns
import matplotlib.pylab as plt

x=["a","b","c","d","e","f"]
y=[1,2,3,4,5,6]
h=[0,0,-1,-1,+1,+1]
s=[10,20,40,10,20,40]
sns.scatterplot(x=x, y=y, hue=h, size=s, legend="full")
plt.legend(labels=["-","/","+","0","1","2"], ncol=2, title="H_Type   S_Type")

Fig. 2

凡例の - と 0 のマーカーが + と 2 のマーカーと同じものになっています。
どうすれば修正できますか？

※新たなる問題。
※s や h の値によって失敗する例。

import seaborn as sns
import matplotlib.pylab as plt

x=["a","b","c","d","e","f"]
y=[1,2,3,4,5,6]
h=[0,0,0,0,0,0]
s=[10,20,10,10,20,20]
sizes = {x: x for x in s}
colp = {-1:'#1F3FDF', 0:'#222222', +1:'#DF1F3F'}
ax=sns.scatterplot(x=x, y=y, hue=h, size=s, sizes=sizes, palette=colp, legend="full")
handles, labels = ax.get_legend_handles_labels()
ax.legend(handles=handles, labels=["-","/","+","0","1","2"], ncol=2, title="H_Type   S_Type")

Answer 1

legend()でhandlesも指定してやる必要があります。
handlesは Axes.get_legend_handles_labels() で現時点の値を取得して、それを使います。

import seaborn as sns
import matplotlib.pylab as plt

x=["a","b","c","d","e","f"]
y=[1,2,3,4,5,6]
h=[0,0,-1,-1,+1,+1]
s=[10,20,40,10,20,40]
ax = sns.scatterplot(x=x, y=y, hue=h, size=s, legend="full")
handles, labels = ax.get_legend_handles_labels()
ax.legend(handles=handles, labels=["-","/","+","0","1","2"], ncol=2, title="H_Type   S_Type")

---

追加の質問に関して。
存在しないデータのlegendを追加しないといけないので、まじめにやるなら https://matplotlib.org/stable/gallery/text_labels_and_annotations/custom_legends.html あたりを参考にして、対応するマーカーを作って、handlesに設定してやればできます。

簡単にするなら、ダミーデータを追加してプロットすれば、legendに出せるようになります。

import seaborn as sns
import matplotlib.pylab as plt

x=["a","b","c","d","e","f"]
y=[1,2,3,4,5,6]
h=[0,0,0,0,0,0]
s=[10,20,10,10,20,20]

x2 = x + [None] * 3
y2 = y + [None] * 3
h2 = h + [-1, 0, 1]
s2 = s + [10, 20, 40]
sizes = {x: x for x in s2}
colp = {-1:'#1F3FDF', 0:'#222222', +1:'#DF1F3F'}
ax = sns.scatterplot(x=x2, y=y2, hue=h2, size=s2, sizes=sizes, palette=colp, legend="full")
handles, labels = ax.get_legend_handles_labels()
ax.legend(handles=handles, labels=["-","/","+","0","1","2"], ncol=2, title="H_Type   S_Type")

Comments

[i113]

どうも有り難うございます。
質問文で例示した状態では上手くいったのですが、sやhの条件が変化するとレジェンドが乱れてしまいます。
条件によらず常に同じレジェンドを表示してほしいのですが……

[bsdfan]

追加の質問については、データに存在しないものをlegendに追加したいという内容になってくるので、ちょっと違う対応が必要です。

[i113]

それは、どうすれば良いのでしょうか？

[i113]

どうも有り難うございます。
上手くいきました。