Кластеризация. График. Легенда

пожалуйста, помогите мне узнать, как создать легенду для моего графика. Я провел кластеризацию и получил вот такие результаты:

0 cluster = 23 values, 3 cluster = 5 values, 2 cluster = 1 values,1 cluster = 1 values.

plt.figure(figsize=(13, 8))

area = np.pi * (X[:, 0]) **2
plt.scatter(X[:, 0], X[:,1], s=250, c=labels.astype(np.float), alpha=0.5, label = '2')
plt.scatter(k_means.cluster_centers_[:,0], k_means.cluster_centers_[:, 1], s = 20, c = 'blue' , label = 'centeroid')
plt.legend()

plt.style.use('fivethirtyeight')
plt.title('Clustering', fontsize=16)
plt.xlabel('funding_rounds_amount', fontsize = 16)
plt.ylabel('angle_profit', fontsize = 11)
plt.show()

My graph

Конечно, код графика не мой, поэтому я не могу понять, как сделать так, чтобы, например, кластер 0 соответствовал фиолетовому и так далее


Ответы (0 шт):