Как добавить легенду на основе цвета на рассеянном диаграмме [matplotlib] ⇐ Python

[Anonymous]

Я пытался добавить легенду на основе цвета в графике рассеяния в Matplotlib. Я видел несколько примеров, но они предполагают, что на каждом графике будет связана этикетка, и я не понял, как сделать это без свойства «метки», и только основывать его на цвете. В настоящее время у меня есть следующий код < /p>

plt.subplot(121)
plt.gca().set_title('Female')
survived = female_data[:,1]
pclass = female_data[:,2]
age = female_data[:,5]
label_color = ['r' if i==0 else 'b' for i in survived]
axes = plt.gca()
axes.get_xaxis().set_visible(False)
plt.scatter(age, pclass, color=label_color)
< /code>

, который делает что -то вроде этого, я хочу, чтобы красная и синяя имели одну метку в сюжете. Есть идея?



Подробнее здесь: https://stackoverflow.com/questions/480 ... matplotlib