Seaborn散点图图例未显示

Question

我正在尝试使用以下代码绘制一些数据

from sklearn.datasets import make_blobs
import seaborn as sns
import numpy as np

X, y = make_blobs(n_samples=1000, n_features=2, centers=10, cluster_std=1.0, center_box=(-10.0, 10.0), shuffle=True, random_state=None)

palette = np.array(sns.color_palette("bright", 10))  #Chossing color  
sns.scatterplot(X[:,0],X[:,1],legend='full',c=palette[y])

颜色很漂亮，但是图例缺失。 当我查看文档时，我看到：

如何绘制图例。如果“brief”，则使用数字hue和size变量...

因此，似乎我还需要包括hue参数。 但是当我尝试在以下代码中使用hue参数时，会创建以下图形...

sns.scatterplot(X[:,0],X[:,1],legend='full',hue=y,c=palette[y])

图例已经显示，但颜色不是我想要的。添加hue参数后，似乎覆盖了调色板参数。无论我选择什么调色板，颜色仍然很丑陋...

我的问题是：如何在保持所需颜色的同时显示图例？

- Raven Cheuk

Answer 1

您需要使用palette参数，并使用y值指定色调。

from sklearn.datasets import make_blobs
import seaborn as sns
import matplotlib.pyplot as plt

X, y = make_blobs(n_samples=1000, n_features=2, centers=10, cluster_std=1.0,
                  center_box=(-10.0, 10.0), shuffle=True, random_state=None)

palette = sns.color_palette("bright", 10)  #Choosing color
sns.scatterplot(X[:, 0], X[:, 1], palette=palette, hue=y, legend='full')
plt.show()

- Troy

---

所以，我在 matplotlib 散点图中混淆了 c kwarg 和 seaborn 的 palette kwarg？现在它可以工作了，谢谢。 - Raven Cheuk