Python итеративный разброс Pyplot Colorbar - PullRequest
0 голосов
/ 08 марта 2019

Я новичок в 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()

Мне нужно установить цветовую шкалу.Я попытался добавить его после, но он нуждается в массиве mapable.Тем не менее, я не могу установить его, когда я вызываю scatter (с этим c = somearray).Также я попытался сохранить точки в отдельных массивах и сделать один разброс, но это также не сработало, потому что я потерял много очков.Я не могу назвать imgshow, потому что он не будет отображать все точки мая также.Моя цель - отобразить градиент для всех отображаемых точек или какую-либо маркировку.Заранее спасибо!

...