如何给Matplotlib imshow绘图的colorbars添加标签？|极客笔记

如何给Matplotlib imshow绘图的colorbars添加标签？

在使用Matplotlib绘图时，经常需要添加colorbar，以方便观察数据的变化。而对于imshow绘图的colorbar来说，我们可以通过一些简单的操作来为colorbars添加标签，方便我们更好地理解图像所代表的数据。

1. 绘制imshow图像

首先，我们需要使用Matplotlib的imshow函数来绘制图像，并将图像以数组的形式输入函数。例如，我们现在要绘制一个10×10的矩阵，并显示出来，代码如下：

import numpy as np
import matplotlib.pyplot as plt

data = np.random.random((10, 10))
plt.imshow(data)
plt.colorbar()

plt.show()

运行该代码后，我们可以看到生成的imshow图像，其中默认会添加一个colorbar用于展示数据的变化。

2. 给colorbar添加标签

为了给colorbar添加标签，我们需要使用Matplotlib的colorbar函数，并在函数中指定标签的内容。例如，我们现在要给上面的图像添加一个名为“Data Value”的标签，代码如下：

import numpy as np
import matplotlib.pyplot as plt

data = np.random.random((10, 10))
imshow = plt.imshow(data)
colorbar = plt.colorbar(imshow)
colorbar.set_label('Data Value')

plt.show()

运行该代码后，我们可以看到新生成的图像，其中colorbar成功添加了标签“Data Value”。

3. 调整colorbar标签的位置和字体大小

如果我们希望将colorbar标签的位置移动到其他地方，或者改变其字体大小，可以使用Matplotlib的colorbar函数中的一些参数来实现。例如，我们现在要把标签位置移动到colorbar图像的顶部，并将字体大小设置为14，代码如下：

import numpy as np
import matplotlib.pyplot as plt

data = np.random.random((10, 10))
imshow = plt.imshow(data)
colorbar = plt.colorbar(imshow, location='top')
colorbar.set_label('Data Value', fontsize=14)

plt.show()

运行该代码后，我们可以看到新生成的图像，其中colorbar成功移动到了顶部，并且字体大小也被改为了14。