xarray contourf绘图中如何让色标刻度标签对较小数值也显示科学计数法

首先可以明确：该需求可以实现，且完全可以通过cbar_kwargs参数完成配置。xarray的绘图接口会将cbar_kwargs内的所有参数直接透传给底层matplotlib的colorbar方法，你可以通过配置色条的刻度格式化参数实现任意阈值下的科学计数法展示。

方案1：强制所有刻度显示为科学计数法 #

如果你希望无论数值大小，色条刻度统一用科学计数法展示，直接在cbar_kwargs中指定format参数为科学计数法格式字符串即可：

import xarray as xr
import numpy as np

# 构造示例测试数据
data = xr.DataArray(
    np.random.randn(10,10)*1e4,
    dims=["x", "y"],
    coords={"x": range(10), "y": range(10)}
)

# 绘图并配置色条
data.plot.contourf(
    cbar_kwargs={
        "format": "%.1e",  # 保留1位小数的科学计数法格式，可根据需求调整小数位数
        "label": "你的数值单位"
    }
)

方案2：自定义阈值触发科学计数法 #

如果你希望仅当数值达到指定阈值（如1e4）时才自动切换为科学计数法，可以搭配matplotlib的ScalarFormatter实现：

import xarray as xr
import numpy as np
from matplotlib.ticker import ScalarFormatter

# 构造示例测试数据
data = xr.DataArray(
    np.random.randn(10,10)*1e4,
    dims=["x", "y"],
    coords={"x": range(10), "y": range(10)}
)

# 定义刻度格式化器
cbar_formatter = ScalarFormatter(useMathText=True)
# 设置触发阈值：绝对值>=1e4 或 <=1e-4时使用科学计数法，可自定义两个阈值
cbar_formatter.set_powerlimits((-4, 4))

# 绘图并配置色条
data.plot.contourf(
    cbar_kwargs={
        "format": cbar_formatter,
        "label": "你的数值单位"
    }
)

提示：将useMathText设为True时科学计数法会显示为上标格式，视觉效果更美观，不需要可改为False使用普通的e表示格式。

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