Python matplotlib Colorfunction

Question

我想在Mathematica中使用类似ColorFunction的{​​{1}}作为我在python中的情节。

换句话说，我想调用pyplot.plot(x, y, color=c)，其中c是一个向量，定义每个数据点的颜色。

有没有办法用matplotlib库实现这个目的？

Answer 1

据我所知，Matplotlib中没有等效物，但我们可以通过两个步骤获得类似的结果：绘制具有不同颜色的点并绘制线条。

这是一个演示。

源代码，

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

fig, ax = plt.subplots()

nrof_points = 100 
x = np.linspace(0, 10, nrof_points)
y = np.sin(x)
colors = cm.rainbow(np.linspace(0, 1, nrof_points))     # generate a bunch of colors

# draw points
for idx, point in enumerate(zip(x, y)):
    ax.plot(point[0], point[1], 'o', color=colors[idx], markersize=10)

# draw the line
ax.plot(x, y, 'k')
plt.grid()

plt.show()