python - 如何在散点图中添加基于颜色的图例 [matplotlib]

Question

我一直在尝试根据 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)

渲染类似这样的东西，我希望红色和蓝色在图中有一个标签。任何想法？

Answer 1

https://colab.research.google.com/notebook#fileId=1RtdwW8ztZbBUpd67RWffKRVt0ae0dNkr

import matplotlib.pyplot as plt
import numpy as np

ages = np.array([1, 2, 1, 3, 1, 2, 2, 1, 1, 2])
pclass = np.arange(10)
survived = np.array([
    True, True, True, False, False,
    False, True, False, True, False
])
plt.plot(
    pclass[survived], ages[survived], 'bo',
    pclass[~survived], ages[~survived], 'ro'
)
plt.legend(['survived', 'died'])

希望您能看到我的 Colab 笔记本或图像。这是你想要的吗？我认为您不想分散。