Python Matplotlib : plot with 2-dimensional arguments : how to specify options?

Question

我正在绘制如下几条曲线:

import numpy as np
import matplotlib.pyplot as plt

plt.plot(x, y)

其中 x 和 y 是二维的(在本例中假设为 N x 2)。

现在我想独立设置每条曲线的颜色。我试过这样的事情:

plot(x, y, color= colorArray)

例如colorArray= ['red', 'black']，但无济于事。其他选项(线型、标记等)也是如此。

我知道这可以通过 for 循环来完成。然而，由于此 plot 命令接受多维 x/y，我认为应该也可以通过这种方式指定绘图选项。

这可能吗？执行此操作的正确方法是什么？ (我在搜索时发现的所有内容都有效地使用了循环)

Answer 1

你可以使用 ax.set_color_cycle :

import numpy as np
import matplotlib.pyplot as plt
np.random.seed(2013)
N = 10
x, y = np.random.random((2,N,2))
x.cumsum(axis=0, out=x)
y.cumsum(axis=0, out=y)
fig, ax = plt.subplots()
colors = ['red', 'black']
ax.set_color_cycle(colors)
ax.plot(x,y)
plt.show()

产量