Python matplotlib Colorfunction

Question

Я хотел бы использовать ColorFunction, аналогичный Mathematica для моих участков в 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()