如何在python中自定义颜色条？

Question

使用此代码，我不知道如何自定义颜色栏。这个网站上的颜色图不能满足我。

shade = m.contourf(Lon,Lat,TBB,np.arange(-90, -20, 10),extend='both',cmap=plt.cm.get_cmap('jet'))       
m.colorbar(shade)

我想用明显的颜色条绘制这样的图片。所以我该怎么做？

Answer 1

您可以使用matplotlib.colors.LinearSegmentedColormap()或定义自己的颜色图matplotlib.colors.ListedColormap()并将其用于您的绘图。

例子：

import numpy as np; np.random.seed(0)
import matplotlib.pyplot as plt
import matplotlib.colors

x = np.arange(0,25)
a = np.random.randint(0,130, size=(25,25))-115
a = np.sort(a).reshape(25,25)

colors = ["#eaa941", "#efef39", "#53a447", "#3b387f", "#48a2ba"]
cmap= matplotlib.colors.ListedColormap(colors)
cmap.set_under("crimson")
cmap.set_over("w")
norm= matplotlib.colors.Normalize(vmin=-100,vmax=-0)

fig, ax = plt.subplots()
im = ax.contourf(x,x,a, levels=[-100,-80,-60,-40,-20,0],
                 extend='both',cmap=cmap, norm=norm)

fig.colorbar(im, extend="both")

plt.show()