如何创建彩色散点图

Question

我想创建一个彩色散点图，我看到了

的用法

plt.scatter(x, y, c = 'rby')

但是当我使用它时，它不起作用。

ValueError: 'c' argument must be a color, a sequence of colors, or a sequence of numbers, not 'rby'

那么如何通过matplotlib制作彩色散点图

Answer 1

如果你想用红色、蓝色和黄色交替散布点，你应该提供一个与你的数据长度相同的颜色列表。

N = len(x)
colors = (list('rby')*(1+N//3))[:N] #messy way of getting ['r', 'b', 'y', 'r',...]
plt.scatter(x, y, c = colors )

How can I create a colorful scatter plot