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设置图形的颜色

1 个答案: