Я новичок в 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, потому что он не будет отображать все точки мая также.Моя цель - отобразить градиент для всех отображаемых точек или какую-либо маркировку.Заранее спасибо!