如何固定Matplotlib contourf色标范围并解决循环绘图色条重复问题

我来帮你搞定这两个循环绘图时的常见小问题，咱们一步步拆解解决：

1. 固定色标范围（-5.0到9.0） #

contourf默认会根据每一批数据的极值自动适配颜色范围，导致不同图的色标差异大。你只需要在调用contourf时手动指定vmin和vmax参数，强制所有图像使用统一的颜色区间即可。

2. 解决色条逐次递增的问题 #

你现在的代码每次循环都在同一个画布上叠加内容，色条自然会越积越多。解决思路是让每个循环对应独立的画布，有两种简单方式：

• 每次循环开始时新建一个画布（plt.figure()）
• 或者每次循环结束后清除当前画布内容（plt.clf()）
  另外注意你代码里的plt.colormap()是笔误，应该是plt.colorbar()，不然色条根本无法正常生成。

修改后的完整代码 #

from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt
import numpy as np  # 补充你遗漏的numpy导入
from numpy import linspace, meshgrid

i = 0
while i < len(inputdata):
    # 每次循环新建独立画布，彻底避免内容叠加
    plt.figure()
    map = Basemap(projection='cyl', llcrnrlat=5.125, llcrnrlon=60.125, urcrnrlat=34.875, urcrnrlon=94.875)
    data = np.array(inputdata[i])
    x = linspace(map.llcrnrx, map.urcrnrx, data.shape[1])
    y = linspace(map.llcrnry, map.urcrnry, data.shape[0])
    xx, yy = meshgrid(x, y)
    # 指定vmin/vmax固定色标范围，同时保存contourf结果用于关联色条
    cs = map.contourf(xx, yy, data, cmap='summer_r', vmin=-5.0, vmax=9.0)
    # 生成对应当前图的色条
    plt.colorbar(cs)
    # 给文件名加索引，避免后续图像覆盖之前的结果
    plt.savefig(f'filename_{i}.jpg', dpi=300)
    # 关闭画布释放内存，尤其数据量大时很重要
    plt.close()
    i += 1

关键修改点说明 #

• plt.figure()：每次循环创建全新画布，从根源上杜绝绘图内容和色条的叠加
• vmin=-5.0, vmax=9.0：强制所有图像使用统一的颜色范围，保证可视化的一致性
• 修正plt.colormap()为plt.colorbar(cs)：正确生成与当前contourf匹配的色条，避免无效调用
• plt.close()：循环结束后关闭画布，防止内存占用过高
• filename_{i}.jpg：通过索引区分不同批次的图像，避免文件被覆盖

如果你不习惯新建画布，也可以把plt.figure()换成plt.clf()放在循环开头，效果是一样的，不过新建画布的方式更直观，不容易出错。

内容的提问来源于stack exchange，提问作者navajyothmp