Python的点图，每个不同的颜色？

Question

python 中的 Matplotlib 采用 x 和 y 值的序列来生成数据图。

我们可以为正在绘制的特定图形指定颜色。但是，由于我们正在绘制一个明确的点列表，我想知道是否有一种方法可以单独指定每个点的颜色？

例如，如何使下图中绘制的每个点都具有特定的随机颜色？

import matplotlib.pyplot as plt
x = np.linspace(0, 2 * np.pi, 400)
y = np.sin(x ** 2)
fig, ax = plt.subplots()
ax.plot(x, y)
ax.set_title('A single color plot')

Answer 1

根据cmets中ImportanceOfBeingErnest的链接和建议，针对需要绘制不同颜色线段的情况，我拼凑了以下解决方案。

理想情况下，我希望每个点都具有特定的颜色，并且线段可以在这些各自颜色之间的色谱中平滑插值。但我想，matplotlib 是免费软件，所以它没有最花哨的功能也就不足为奇了。除非，当然，如果可以做到，但我不知道怎么做？

import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection
x = np.linspace(0, 2 * np.pi, 400)
y = np.sin(x ** 2)
xy = np.array([x, y]).T.reshape(-1, 1, 2)
segments = np.hstack([xy[:-1], xy[1:]])
fig, ax = plt.subplots()
lc = LineCollection(segments, colors=np.random.rand(len(segments), 3))
ax.add_collection(lc)
ax.autoscale()
ax.set_title('A multi-color plot')
plt.show()

编辑：

只是为了好玩，做更多的步骤并“手动”创建渐变，正如ImportanceOfBeingErnest 建议的那样，我们可以得到例如像这样：

import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection
x = [0.]
while x[-1]<2 * np.pi :
    x.append(x[-1] +1/(100*np.sqrt(1+4*x[-1]*x[-1]*np.cos(x[-1]*x[-1])*np.cos(x[-1]*x[-1]))))
x = np.array(x)
y = np.sin(x ** 2)
xy = np.array([x, y]).T.reshape(-1, 1, 2)
segments = np.hstack([xy[:-1], xy[1:]])
myColors = np.random.rand(len(segments), 3)
for i in range(len(segments)//10):
    for j in range(9):
        myColors[10*i+j+1] = myColors[10*i]+(myColors[10*(i+1)]-myColors[10*i])*(j+1)/10
fig, ax = plt.subplots()
lc = LineCollection(segments, colors=myColors)
ax.add_collection(lc)
ax.autoscale()
ax.set_title('A multi-color plot')
plt.show()