python matplotlib自定义colorbar颜色条及内置色条详解

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

自定义colorbar（draw colorbar without any mapple/plot）

参考：Customized Colorbars Tutorial

api example code: colorbar_only.py

自定义colorbar可以画出任何自己想要的colorbar，自由自在、不受约束，不依赖于任何已有的图(plot/mappable)。这里使用的是mpl.colorbar.ColorbarBase类，而colorbar类必须依赖于已有的图。

关于该类的参数如下 (该类参数描述没找到，部分参数的取值可参考colorbar中的描述)：

class matplotlib.colorbar.ColorbarBase(ax, cmap=None, norm=None, alpha=None, values=None, boundaries=None, orientation=‘vertical’, ticklocation=‘auto’, extend=‘neither’, spacing=‘uniform’, ticks=None, format=None, drawedges=False, filled=True, extendfrac=None, extendrect=False, label=’’)[source]
参数简单描述

ax :可用于设置colorbar的位置、长、宽
norm :用于规范化–设置颜色条最大最小值
cmap：颜色（可参考本篇博文的最后部分——推荐色带与自定义色带）
boundaries：要想使用extend，在norm之外，必须要有两个额外的boundaries
orientation：colorbar方向，躺平or垂直
extend：延伸方向(在norm之外colorbar可延伸)
ticks：自定义各段的tick（记号）

下面是一个例子，可以自己试试调整各项参数：

import numpy as np
import pylab as plt
import matplotlib as mpl

if __name__ == '__main__':
	fig = plt.figure(figsize=(3, 8))
	cmap = mpl.cm.Spectral_r
	ax3 = fig.add_axes([0.3, 0.2, 0.2, 0.5]) # 四个参数分别是左、下、宽、长
	norm = mpl.colors.Normalize(vmin=1.3, vmax=2.5)
	bounds = [ round(elem, 2) for elem in np.linspace(1.3, 2.5, 14)] # 
	cb3 = mpl.colorbar.ColorbarBase(ax3, cmap=cmap,
								norm=norm,
								# to use 'extend', you must
								# specify two extra boundaries:
								boundaries= [1.2] + bounds + [2.6],
								extend='both',
								ticks=bounds,  # optional
								spacing='proportional',
								orientation='vertical')
	plt.show()

contourf函数

contour和contourf分别绘制轮廓线和填充轮廓线。两个版本的函数签名和返回值是相同的。关于两者画图的一些实例可以参阅python matplotlib contour画等高线图，以及matplotlib中文文档

    contourf([X, Y,] Z, [levels], **kwargs) # [ ]表示该参数是可选的——可以给定也可以不给定

contourf中的基本参数：

X, Y : array-like, optional

contourf参数中X和Y都必须是二维的，形状与Z相同(例如，通过numpy.meshgrid创建)；或者都必须是一维的，这样len(X) == M是Z中的列数，len(Y) == N是Z中的行数。
如果没有给定，则假设它们是整数索引，即X = range(M)， Y = range(N)。

Z : array-like(N, M)

绘制轮廓线的高度值。

levels : int or array-like, optional

确定等高线/区域的数量和位置。
如果是int n，则使用n个数据间隔;即画n+1条等高线。水平高度是自动选择的。
如果类似数组，则在指定的级别上绘制等高线。这些值必须是递增的。

为所有子图添加一个色带

参考：Matplotlib 2 Subplots, 1 Colorbar

在有图的情况下，为所有子图添加一个colorbar：

import numpy as np
import matplotlib.pyplot as plt

fig, axes = plt.subplots(nrows=2, ncols=2)
for ax in axes.flat:
    im = ax.imshow(np.random.random((10,10)), vmin=0, vmax=1)

fig.subplots_adjust(right=0.8)
cbar_ax = fig.add_axes([0.85, 0.15, 0.05, 0.7])
fig.colorbar(im, cax=cbar_ax)

plt.show()