自定义colorbar

要自定义colorbar，您可以使用Matplotlib库中的colorbar()函数。该函数可以接受一个颜色映射对象，以及一些其他参数，以创建您自己喜欢的颜色条。

下面是一个简单的示例代码，用于创建一个自定义颜色条：

import matplotlib.pyplot as plt
import numpy as np

# Generate some data for plotting
x = np.linspace(-10, 10, 101)
y = np.linspace(-10, 10, 101)
xx, yy = np.meshgrid(x, y)
z = np.sqrt(xx**2 + yy**2)

# Set up the plot
fig, ax = plt.subplots()

# Create a custom colormap
cmap = plt.cm.get_cmap('cool')
new_cmap = cmap(np.linspace(0.25, 0.75, 256))
new_cmap[:1,:] = np.array([1,1,1,1]) # Set the first row of the colormap to white
new_cmap[-1:,:] = np.array([0,0,0,1]) # Set the last row of the colormap to black
new_cmap = plt.colors.ListedColormap(new_cmap)

# Plot the data with the custom colormap
im = ax.imshow(z, cmap=new_cmap, vmin=0, vmax=20)

# Create the colorbar
cbar = fig.colorbar(im, ax=ax)

# Show the plot
plt.show()

在这个例子中，我们首先生成一些数据，并使用Matplotlib的imshow()函数将其绘制到图表上。然后，我们使用get_cmap()函数创建一个名为“cool”的颜色映射对象，并使用np.linspace()函数将其缩小到0.25到0.75的范围内。接下来，我们将该颜色映射的第一行设置为白色，将最后一行设置为黑色，并将其转换为一个列表颜色映射对象。最后，我们使用colorbar()函数创建颜色条，并将其添加到图表上。

您可以根据自己的需要调整颜色映射对象的参数，以创建您自己喜欢的颜色条。