Категориальный график по цвету - PullRequest
1 голос
/ 29 мая 2020

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

    [![fig = plt.figure(figsize = (20,10))
ax = fig.add_subplot(1,1,1) 
ax.set_xlabel('Sentiment Score', fontsize = 15)
ax.set_ylabel('Star Review', fontsize = 15)
ax.set_title('Heinz Segmentation', fontsize = 20)


targets = \['Sugar/ Healthy Positives', 'Sugar/ Healthy Negatives', 'Price/ Value Positives', 'Price/ Value Negatives', 'Purists Positives', 'Purists Negatives'\]
colors = \['r', 'g', 'b', 'c', 'm', 'y'\]
for target, color in zip(targets,colors):
    ax.scatter(df.Compound,df.StarsInt
               , c = color
               , s = 200
              ,alpha = .5)
ax.legend(targets, prop={'size': 14})
ax.grid()][1]][1]

1 Ответ

1 голос
/ 29 мая 2020

IIU C, у вас проблема в легендах. Вы забыли передать легенду при построении

for target, color in zip(targets,colors):
    ax.scatter(df.Compound,df.StarsInt, c=color,
               s = 200,alpha = .5,       
               label=target) # <--- pass the label to show in legend
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...