流图的颜色范围

Question

我需要创建一个带有颜色条的流图，指示给定线的速度。问题是，我的速度数据中有 2 个点的值过高，完全重新调整了颜色条的比例，而且我无法真正删除它们。

我已经搜索了在不修改数据的情况下阻止它的方法（例如 cmin、Vmin、Vmax...），但似乎 Streamplot 没有这方面的功能...

所以有没有办法添加限制到流图中的 cmap ？

Answer 1

我认为这应该有帮助。

## import matplotlib 
matplotlib as mpl

## Set colormap for the colorbar
cmap = mpl.cm.rainbow

#### Use norm to set you vmax and vmin
norm = mpl.colors.Normalize(vmin=0, vmax=35)

## Pass norm as argument while plotting
strm = ax.streamplot(x=ds["longitude"],y=ds["latitude"], u=ds["u"], v=ds["v"], color=ds.u.values, linewidth=1, cmap='rainbow', density=0.6, norm=norm)

fig.colorbar(strm.lines, label = 'wind speed [m$^2$]', shrink=0.6)