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

[Anonymous]

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

Код: Выделить всё

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)

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




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