python - 如何标记散点图

Question

我想根据值 1 或 0 制作具有两种颜色的散点图。

plt.figure(figsize=(15,5))
x = np.linspace(0, 12, 12)
y = np.random.random(12)
z = [0, 0, 1, 1, 0, 0, 1, 1, 0, 0, 1, 1]
plt.scatter(x, y, c=z, label=z)
plt.legend()

然后看起来像这样： 我怎样才能让图例用紫色和黄色说 0 和 1？

更好的是让图例说出文本，例如“零”和“一”。似乎只有 1 个可以标记的东西，我在传说中再也没有得到更多的点。

我希望我足够清楚以了解问题所在。

标签： pythonlabelscatter