matplotlib图上的不同背景颜色区域

Question

我正在绘制一个随时间变化的变量，我想在交通灯样式中指出变量的严重性。

 |
 |
 |           red
 |
 |
y|              /
 |             /   yellow
 |      ___   /
 |     /   \_/
 |    /           green
 |   /
 |_____________________________
             x

因此，对于低y值，背景为绿色，中间为黄色/琥珀色，高值为红色。最终看起来有点像三条纹旗帜。

由于我经常重新绘制这个并且y比例可能会改变，所以我变得更加复杂，所以我不能在图上放置固定的盒子，除非我计算我每次绘图时的盒子大小。

有没有办法用matplotlib做到这一点？

我目前的密码：

import matplotlib.pyplot as plt

plt.plot(x, y, 'k')
plt.grid(b=True)
ax = plt.gca()
ax.set_facecolor("some colour")

Answer 1

您可以使用ax.axhspan()轻松完成此操作。您将必须传递您想要着色的区域的y坐标（这是一个手动过程）

import matplotlib.pyplot as plt

x = [1,2,3,4,5,6,7,8]  #fake data
y = [1,2,3,4,3,2,9,12]

fig, ax = plt.subplots()

ax.plot(x, y, 'k')
ax.grid()
ax.margins(0) # remove default margins (matplotlib verision 2+)

ax.axhspan(0, 4, facecolor='green', alpha=0.5)
ax.axhspan(4, 9, facecolor='yellow', alpha=0.5)
ax.axhspan(9, 12, facecolor='red', alpha=0.5)

plt.show()

给出了：