python - 在 matplotlib 上的散点图中为每个系列设置不同的颜色

Question

假设我有三个数据集:

X = [1,2,3,4]
Y1 = [4,8,12,16]
Y2 = [1,4,9,16]

我可以散点图:

from matplotlib import pyplot as plt
plt.scatter(X,Y1,color='red')
plt.scatter(X,Y2,color='blue')
plt.show()

我怎样才能用 10 套做到这一点？

我搜索了这个，可以找到任何关于我所问内容的引用。

编辑:澄清(希望)我的问题

如果我多次调用 scatter ，我只能在每个 scatter 上设置相同的颜色。另外，我知道我可以手动设置颜色数组，但我确信有更好的方法来做到这一点。我的问题是，“我怎样才能自动散点图我的几个数据集，每个数据集都有不同的颜色。

如果有帮助，我可以轻松地为每个数据集分配一个唯一编号。

Answer 1

我不知道您所说的“手动”是什么意思。您可以轻松地选择颜色图并制作颜色数组:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm

x = np.arange(10)
ys = [i+x+(i*x)**2 for i in range(10)]

colors = cm.rainbow(np.linspace(0, 1, len(ys)))
for y, c in zip(ys, colors):
    plt.scatter(x, y, color=c)

或者您可以使用 itertools.cycle 制作自己的颜色循环器，并指定要循环的颜色，使用 next 来获得所需的颜色。例如，有 3 种颜色:

import itertools

colors = itertools.cycle(["r", "b", "g"])
for y in ys:
    plt.scatter(x, y, color=next(colors))

想一想，也许不使用 zip 和第一个更清洁:

colors = iter(cm.rainbow(np.linspace(0, 1, len(ys))))
for y in ys:
    plt.scatter(x, y, color=next(colors))