6

新的 matplotlib 用户在这里。我正在尝试绘制颜色编码的数据行，或者更好的是，绘制颜色编码的数据范围。沿 y 轴的颜色编码间隔。粗略的演示脚本如下：

import matplotlib.pyplot as plt

# dummy test data
datapoints = 25
maxtemps = [ 25, 24, 24, 25, 26, 27, 22, 21, 22, 19, 17, 14, 13, 12, 11, 12, 11, 10, 9, 9, 9, 8, 9, 9, 8 ]
mintemps = [ 21, 22, 22, 22, 23, 24, 18, 17, 16, 14, 10, 8, 8, 7, 7, 6, 5, 5, 5, 4, 4, 4, 3, 4, 3 ]
times = list(xrange(datapoints))

# cap a filled plot at a given level
def capped(indata, cap):
    outdata  = [0] * datapoints
    lcount = 0

    while lcount < datapoints:
        if indata[lcount] > cap:
            outdata[lcount] = cap
        else:
            outdata[lcount] = indata[lcount]
        lcount += 1
    return outdata

fig = plt.figure()

ax1 = fig.add_subplot(111)
ax1.fill_between(times, 0, maxtemps, color='#FF69B4', zorder=1, linewidth=0.1)
ax1.fill_between(times, 0, capped(maxtemps,25), color='#F08228', zorder=2, linewidth=0.1)
ax1.fill_between(times, 0, capped(maxtemps,20), color='#E6AF2D', zorder=3, linewidth=0.1)
ax1.fill_between(times, 0, capped(maxtemps,15), color='#E6DC32', zorder=4, linewidth=0.1)
ax1.fill_between(times, 0, capped(maxtemps,10), color='#A0E632', zorder=5, linewidth=0.1)
ax1.fill_between(times, 0, capped(maxtemps,5), color='#00DC00', zorder=6, linewidth=0.1)
ax1.fill_between(times, 0, mintemps, color='#FFFFFF', zorder=7, linewidth=0.1)
plt.setp(ax1.get_xticklabels(), visible=False)
ax1.grid(True, zorder=8)
ylim(0)
plt.draw()
plt.show()

完成了大部分工作，但它引出了两个问题。

  1. 有没有一种更直接、更优雅的方式来使用我不知道的 matplotlib 功能来实现同样的效果？也就是说，要么绘制一个（比如说）时间序列数据的一维数组，要么说明两组此类数据之间的范围（例如，最高、最低温度）？

  2. 尽我所能，我无法说服网格线绘制在图形的顶部。它们似乎总是位于第一组绘制数据的顶部，然后被后续绘制的数据所掩盖，而下半部分则为空白。zorder 的使用似乎被忽略了。

非常感谢。

4

1 回答 1

9

回答你的第二个问题：

您可以将所有fill_betweenzorder 设置为 0.1、0.2、0.3 ...

网格线属于xaxis和yaxis，xaxis和yaxis的zorder为2.5。因此任何小于 2.5 的 zorder 都会显示在网格线下方。

我写了一些与你类似的代码，但使用 for 循环和numpy.interp,numpy.clip来做剪辑图：

import pylab as pl
import numpy as np
maxtemps = [ 25, 24, 24, 25, 26, 27, 22, 21, 22, 19, 17, 14, 13, 12, 11, 12, 11, 10, 9, 9, 9, 8, 9, 9, 8 ]
mintemps = [ 21, 22, 22, 22, 23, 24, 18, 17, 16, 14, 10, 8, 8, 7, 7, 6, 5, 5, 5, 4, 4, 4, 3, 4, 3 ]
times = list(xrange(len(mintemps)))

colors = [
    (25, '#FF69B4'),
    (20, '#F08228'),
    (15, '#E6AF2D'),
    (10, '#E6DC32'),
    (5,  '#A0E632'),
    (0,  '#00DC00')
]

# change 300 to larger number if you need more accuracy.
x = np.linspace(times[0], times[-1], 300) 
maxt = np.interp(x, times, maxtemps)
mint = np.interp(x, times, mintemps)

last_level = np.inf
for level, color in colors:    
    tmp_min = np.clip(mint, level, last_level)
    tmp_max = np.clip(maxt, level, last_level)
    pl.fill_between(x, tmp_min, tmp_max, lw=0, color=color)
    last_level = level

pl.grid(True)
pl.show()

在此处输入图像描述

于 2012-04-27T01:34:39.653 回答