如何在Matplotlib中调整色标范围并适配NOAA海洋硝酸盐数据

问题描述 #

我在绘制NOAA下载的海洋表层硝酸盐数据可视化图时，遇到色标调整问题：想要把色标范围设为0-8，所有大于8的值用色谱顶端颜色表示，同时保留绘图的其他元素。但使用.colorbar参数时会生成单独的空白色标，需要可行的解决方案。

可复现代码 #

import cartopy.crs as ccrs
import cartopy
from matplotlib import pyplot as plt
import cartopy.feature as cfeature
import numpy as np
import xarray as xr

nit = 'https://www.ncei.noaa.gov/thredds-ocean/dodsC/ncei/woa/nitrate/all/1.00/woa18_all_n00_01.nc'

nitr= xr.open_dataset(nit, decode_times=False)
nitr

variables = ['n_an']
nitr[variables]

nitr_5 = nitr.n_an.sel(depth=5,  method='nearest')
nitr_5

fig = plt.figure(figsize=(9,6))
ax = plt.axes(projection=ccrs.Robinson())
ax.coastlines()
ax.gridlines()
nitr_5.plot(ax=ax, transform=ccrs.PlateCarree(), vmin=2, vmax=30, cbar_kwargs={'shrink': 0.4}, cbar)
ax.set_title("")

解决方案 #

修改后的完整代码 #

import cartopy.crs as ccrs
from matplotlib import pyplot as plt
import xarray as xr

# 加载NOAA硝酸盐数据集
nit = 'https://www.ncei.noaa.gov/thredds-ocean/dodsC/ncei/woa/nitrate/all/1.00/woa18_all_n00_01.nc'
nitr = xr.open_dataset(nit, decode_times=False)

# 提取5米深度的硝酸盐异常值数据
nitr_5 = nitr.n_an.sel(depth=5, method='nearest')

# 创建绘图画布与投影轴
fig = plt.figure(figsize=(9,6))
ax = plt.axes(projection=ccrs.Robinson())
ax.coastlines()
ax.gridlines()

# 绘制数据并配置色标
nitr_5.plot(
    ax=ax,
    transform=ccrs.PlateCarree(),
    vmin=0,          # 色标下限设为0
    vmax=8,          # 色标上限设为8
    extend='max',    # 大于8的值用色谱顶端颜色表示，色标显示超范围标记
    cbar_kwargs={'shrink': 0.4}  # 调整色标大小，避免重复生成空白色标
)
ax.set_title("")
plt.show()

关键修改说明 #

• extend='max'：这是实现超范围值可视化的核心参数，它会将所有大于vmax的值统一显示为色谱顶端颜色，同时在色标右侧添加一个小三角标记，提示存在超出上限的数据。
• 移除多余的cbar参数：原代码中额外传递的cbar参数会触发Matplotlib生成重复的空白色标，所有色标相关配置统一放在cbar_kwargs中即可。
• 色标范围限定：通过vmin=0和vmax=8严格控制色标的数值区间，确保可视化的颜色映射聚焦在0-8的范围内。

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