为什么在这种情况下无法设置固定的颜色条?

时间:2018-10-02 00:30:27

标签: matplotlib clim

我正在尝试使用相同的颜色条绘制一堆极坐标图。但是,在设置打印极限后,颜色条会有所不同。在下面的代码段中,我随机创建了5张地图,但在固定范围内绘制,但输出图形在颜色栏中仍然不同。

from numpy import linspace, pi, ndarray, random
import matplotlib
matplotlib.use('Agg')
from matplotlib.pyplot import figure

lon = linspace(start=0, stop=2*pi, num=100)
colat = linspace(start=0, stop=9, num=10)
emission = ndarray(shape=(10, 100, 5), dtype=float)
for t in range(5):
    emission[:, :, t] = random.rand(10, 100)

fig = figure(num='emission', figsize=(15, 15))
em_pos = [0.05, 0.1, 0.8, 0.8]
emc_pos = [0.9, 0.1, 0.05, 0.8]
for t in range(5):
    fig.clear()

    ax = fig.add_subplot(121, polar=True, position=em_pos)
    axcont = ax.contourf(lon, colat, emission[:, :, t], vmin=0, vmax=2)

    axc = fig.add_subplot(122, position=emc_pos)
    fig.colorbar(mappable=axcont, cax=axc)

    fig.savefig(fname='emission{0:d}.png'.format(t), format='png')

1 个答案:

答案 0 :(得分:0)

问题似乎已解决。这不是颜色条的问题,而是contourf的问题。当我更换

ax.contourf(lon, colat, emission[:, :, t], vmin=0, vmax=2)

使用

ax.pcolormesh(lon, colat, emission[:, :, t], vmin=0, vmax=2)

然后,颜色栏显示适当的范围。确实,这不是一个完整的解决方案,pcolormesh在某些方面与contourf不同,但是可以满足我的需求。