Кластеризация. График. Легенда
пожалуйста, помогите мне узнать, как создать легенду для моего графика. Я провел кластеризацию и получил вот такие результаты:
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()
Конечно, код графика не мой, поэтому я не могу понять, как сделать так, чтобы, например, кластер 0 соответствовал фиолетовому и так далее
