我可以将点的颜色列表传递给 matplotlib 的“Axes.plot()”吗？

Question

我有很多点要绘制，我注意到在 matplotlib 中单独绘制它们比绘制它们需要更长的时间（根据cProfile，超过 100 倍以上）一次全部。

但是，我需要对点进行颜色编码（基于与每个点关联的数据），并且无法弄清楚如何为给定的Axes.plot() 调用绘制多种颜色。例如，我可以得到与我想要的结果类似的结果

fig, ax = matplotlib.pyplot.subplots()
rands = numpy.random.random_sample((10000,))
for x in range(10000):
    ax.plot(x, rands[x], 'o', color=str(rands[x]))
matplotlib.pyplot.show()

但宁愿做一些更快的事情，比如

fig, ax = matplotlib.pyplot.subplots()
rands = numpy.random.random_sample((10000,))
# List of colors doesn't work
ax.plot(range(10000), rands, 'o', color=[str(y) for y in rands])
matplotlib.pyplot.show()

但提供一个列表作为color 的值不能以这种方式工作。

有没有办法向Axes.plot() 提供颜色列表（以及就此而言，边缘颜色、面颜色、形状、z 顺序等），以便可以自定义每个点，但所有点可以一次绘制吗？

---

使用Axes.scatter() 似乎已经成功了，因为它允许单独设置点颜色；但颜色似乎就这么远了。 （Axes.scatter() 的布局也完全不同。）

Answer 1

我直接创建对象（补丁）的速度大约快 5 倍。为了说明示例，我更改了限制（必须使用此方法手动设置）。圆圈本身是用matplotlib.path.Path.circle 绘制的。最小的工作示例：

import numpy as np
import pylab as plt
from matplotlib.patches import Circle
from matplotlib.collections import PatchCollection

fig, ax = plt.subplots(figsize=(10,10))
rands = np.random.random_sample((N,))

patches = []
colors  = []

for x in range(N):
    C = Circle((x/float(N), rands[x]), .01)
    colors.append([rands[x],rands[x],rands[x]])
    patches.append(C)

plt.axis('equal')
ax.set_xlim(0,1)
ax.set_ylim(0,1)

collection = PatchCollection(patches)
collection.set_facecolor(colors)
ax.add_collection(collection)
plt.show()