matplotlib中的colorbar和min、max介绍

在matplotlib库中，plot图表时，有时我们需要添加colorbar来标注颜色的取值范围，也需要获取plot图表的数据的最小值和最大值，以便于对数据进行后续处理。本文将介绍如何在matplotlib中添加colorbar，并获取plot图表的数据的最小值和最大值。

添加colorbar：

通过调用colorbar函数，可以在plot图表旁边添加colorbar。colorbar函数需要传入3个参数：plot图表对象、可选参数cmap、可选参数label。其中，cmap表示颜色映射表，用于指定colorbar要显示的颜色范围。label表示colorbar的标签。

import matplotlib.pyplot as plt
import numpy as np

# 生成数据
x = np.arange(0, 4 * np.pi, 0.1)
y = np.arange(0, 2 * np.pi, 0.1)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) + np.cos(Y)

# 绘制图表
plt.figure()
im = plt.imshow(Z, cmap='coolwarm')

# 添加colorbar
cbar = plt.colorbar(im, label='值范围')

plt.show()

以上代码绘制了一个二维的正弦余弦图，通过colorbar显示了图中数据的颜色范围。具体效果如下图所示：

获取min和max：

对于plot图表的数据，如果需要获取数据的最小值和最大值，可以使用np.min和np.max函数。

import matplotlib.pyplot as plt
import numpy as np

# 生成数据
x = np.arange(0, 4 * np.pi, 0.1)
y = np.arange(0, 2 * np.pi, 0.1)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) + np.cos(Y)

# 绘制图表
plt.figure()
im = plt.imshow(Z, cmap='coolwarm')

# 获取最小值和最大值
Z_min = np.min(Z)
Z_max = np.max(Z)
print('最小值：', Z_min)
print('最大值：', Z_max)

# 添加colorbar
cbar = plt.colorbar(im, label='值范围')

plt.show()

以上代码在绘制图表的同时，获取了图表数据的最小值和最大值。控制台输出的结果如下所示：

最小值： -1.9838235802504072
最大值： 1.9938384644612547

以上就是在matplotlib中添加colorbar，并获取plot图表的数据的最小值和最大值的方法。