Python Matplotlib设置图的颜色

Question

我想使用plt.step()使用plt.step()绘制阶梯图。 我将y值作为np.array的数组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()