如何使中心的颜色使用相同的簇颜色？

Question

我有两种方法用于绘制群集的中心

def draw_roh_delta(self, rho, delta, center, plot):
        plot.scatter(rho, delta, label='rho-delta', c='k', s=5)
        plot.set_xlabel('rho')
        plot.set_ylabel('delta')
        center_rho = rho[center]
        center_delta = delta[center]
        np.random.seed(6)
        colors = np.random.rand(len(center), 3)
        plot.scatter(center_rho, center_delta, c=colors)
        plot.legend()

，还有其他用于绘制群集

def draw_cluster(self, title, cluster, halo, points, plot):
    cluster_points = dict()
    colors = dict()
    np.random.seed(10)
    for k, v in cluster.items():
        cluster_points[k] = points.loc[cluster[k], :]
        colors[k] = np.random.rand(3)
    for k, v in cluster_points.items():
        plot.scatter(v.loc[:, 'x'], v.loc[:, 'y'], c=colors[k], alpha=0.5)
        plot.scatter(v.at[k, 'x'], v.at[k, 'y'], c=colors[k], s=np.pi * 10 ** 2)

的方法，我需要使中心的颜色与簇相同，而不是随机创建它们。有人可以帮忙吗？

编辑

def draw_roh_delta(self, rho, delta, center, plot):
    plot.scatter(rho, delta, label='rho-delta', c='k', s=5)
    plot.set_xlabel('rho')
    plot.set_ylabel('delta')
    center_rho = rho[center]
    center_delta = delta[center]
    color= ['red', 'blue', 'green', 'yellow','orange','purple']
    plot.scatter(center_rho, center_delta, c=color)
    plot.legend()

Answer 1

一种方法是使用绘图函数来绘制点，该函数返回list [line2d]。您只需要取出第一个元素并调用其get_color（）方法即可获得这次绘制的颜色。

我从Kmeans聚类中提取了我的结果：

fig, ax = plt.subplots()

for i, center in enumerate(centers):
    color = ax.plot(*center, '*')[0].get_color()
    points = data_set[labels == i]
    ax.plot(*points.T, 'o', c=color)

不幸的是，要做到这一点，您可能需要将这两种方法融合在一起，或者让第一种方法将所有颜色返回第二种方法。

edit

我会这样解决：

def draw_rho_delta(self, rho, delta, center, plot):
    # ...
    self.colors = []
    for c_rho, c_delta in zip(center_rho, center_delta):
        color = plot.plot(c_rho, c_delta, 'o')[0].get_color()
        self.colors.append(color)

def draw_cluster(self, title, cluster, halo, points, plot):
    # ...
    for (k, v), color in zip(cluster.items(), self.colors):
        p = points.loc[v, :]
        plot.plot(p.loc[:, 'x'], p.loc[:, 'y'], 'o', c=color)
        plot.plot(p.at[k, 'x'], p.at[k, 'y'], 'o', c=color)

我不知道您的plot的类型，所以我假设它是matplotlib.pyplot.axes.axes，而且我认为您绘制中心的顺序与绘制簇的顺序一致。