如何将图例添加到项目依赖于整数变量的pyplot中？

Question

我正在实现一个K-Means聚类算法。我想在散点图中显示聚集的数据集（每个数据集有不同的颜色）。我的做法如下：

for i in range(k):
    plt.scatter(np.array(clustersets[i])[:, 0], np.array(clustersets[i])[:, 1], c=c_map(i))

，其中k是中心数（->；散点图中的簇集数。 我现在想添加一个图例，其中每个数据集包含一个项目。因此，这取决于预定义的数字k

我怎样才能添加一个图例，使其覆盖散点图中的所有不同项目，而不考虑k是什么

Answer 1

你能试试c=KMeans(n_clusters=k).fit(X).labels_astype(float)吗

Answer 2

如果我理解正确，您需要一个从0到k的图例。您可以使用带有fiel说明符的label选项%d来指定数据集

for i in range(k):
    plt.scatter(np.array(clustersets[i])[:, 0], 
                np.array(clustersets[i])[:, 1], c=c_map(i), label='Dataset %d' %i)
plt.legend()