Python迭代散点pyplot颜色栏

Question

我是python和机器学习的新手，我正努力在自己的绘图中添加颜色条。我正在建立这样的情节：

cluster_membership = np.argmax(u, axis=0)
    x = PCA(n_components=2).fit_transform(x).T
    for j in range(k):
        plt.scatter(
            x[0][cluster_membership == j],
            x[1][cluster_membership == j])
    plt.grid(True)
    plt.show()

我需要设置一个颜色栏。我尝试在之后添加它，但是它需要一个可映射的数组。但是，当我调用散点图（使用c = somearray）时，无法设置它。我也尝试将点保存在单独的数组中并进行一次分散，这也没有用，因为我丢失了很多点。我无法调用imgshow，因为它也不会显示所有可能的点。我的目标是为所有显示的点或某种标签显示一个渐变。预先感谢！