Ошибка легенды 3D-графика рассеяния от KMeans Не найдены дескрипторы с метками для добавления в легенду

Я построил трехмерный график рассеяния для модели KMeans, которую я подогнал для RFM-анализа. Я использовал метки моделей KMeans для «цветовых» групп. когда я использовал легенду (), он выдает ошибку: «Нет дескрипторов с метками для добавления в легенду»

from mpl_toolkits.mplot3d import Axes3D     
%matplotlib notebook     

fig = plt.figure(figsize=(8, 6))     
ax = fig.add_subplot(111, projection='3d')      

xs = RFM['Recency'].dt.days      
ys = RFM['Frequency']      
zs = RFM['Value']      
ax.scatter(xs, ys, zs, s=50, alpha=0.6, c=final_model.labels_, cmap='rainbow')      

ax.set_xlabel('Recency(days)')      
ax.set_ylabel('Frequency')      
ax.set_zlabel('Value')      

ax.legend()

python k-means legend scatter3d
person Erandi    schedule 24.09.2019    source источник
comment
Какую легенду вы ожидаете? Возможно, вы путаете его с matplotlib.pyplot.colorbar   -  person Itay    schedule 24.09.2019
comment
Я хочу, чтобы метки клиентов (1,2,3...) были назначены каждому цвету радуги на графике в виде легенды.   -  person Erandi    schedule 24.09.2019
comment
Но как связать метки клиентов с данными (x, y, z)?   -  person Itay    schedule 24.09.2019

Ответы (1)


arrow_upward
1
arrow_downward

Проблема с вашим кодом заключается в том, что вы не указали label для разбросанных точек, поэтому нет причин, по которым legend будет работать.

Следуя руководству по Рассеянные диаграммы с легендой на веб-сайте matplotlib, код рандомизирует данные с тремя «классами» и отображает их с легендой:

from matplotlib import pyplot as plt
import numpy as np

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, projection='3d')

xs = np.random.normal(0, 1, (20,))
ys = np.random.normal(0, 1, (20,))
zs = np.random.normal(0, 1, (20,))

labels = np.random.choice(["First", "Second", "Third"], (20,))

for lbl in np.unique(labels):
    indices = np.where(labels == lbl)
    x = xs[indices]
    y = ys[indices]
    z = zs[indices]
    print(x,y,z,lbl)
    ax.scatter(x, y, z, s=50, alpha=0.6, label=str(lbl), cmap='rainbow')

ax.legend()

plt.show()

Результат:

Вывод кода

person Itay    schedule 24.09.2019
comment
большое спасибо Итай. Каким-то образом этот код оказался для меня ошибкой; Я сделал это с помощью matplotlib.patches. - person Erandi; 24.09.2019