Python matplotlib 设置图表的颜色

Question

我想使用 plt.step() 用 pyplot 绘制阶梯图。我将我的 y 值作为 np.array 的数组传递。现在我想给不同的图上色。我如何传递一组图形颜色，这些颜色应用于图表中的不同图形？

yvalues = [arrayA, arrayB, arrayC]    
ycolors = ["colorA", "colorB", "colorC"]
plt.step(xvalues, yvalues, ycolors)

我只知道按值做：

plt.step(xvalues, arrayA, "colorA", arrayB, "colorB", arrayC, "colorC")

Answer 1

可以通过 for 循环来完成

import itertools
import numpy as np
import matplotlib.pyplot as plt

# your x data
x = np.arange(10)

# your y data
data = np.arange(1,4)[:,None] * np.arange(10)[None,:]

# your colors
color = ['k','r','g']

fig,ax = plt.subplots(1)

for y,c in itertools.izip(data,color):
    ax.step(x,y,c)

plt.show()

Python matplotlib 设置图表的颜色

Python matplotlib set color of graphs

python

matplotlib