Matplotlib：简单颜色条

简介

在本实验中，我们将学习如何在 Matplotlib 绘图中添加一个简单的颜色条。颜色条是一种用于可视化绘图中颜色所代表的值范围的有用工具。我们将使用imshow函数创建一个简单的二维绘图，然后为其添加一个颜色条。

虚拟机使用提示

虚拟机启动完成后，点击左上角切换到笔记本标签页，以访问 Jupyter Notebook 进行练习。

有时，你可能需要等待几秒钟让 Jupyter Notebook 完成加载。由于 Jupyter Notebook 的限制，操作验证无法自动化。

如果你在学习过程中遇到问题，随时向 Labby 提问。课程结束后提供反馈，我们将立即为你解决问题。

导入库

首先，我们需要导入必要的库。在本实验中，我们将使用numpy和matplotlib.pyplot库。

import matplotlib.pyplot as plt
import numpy as np

创建二维绘图

接下来，我们将使用imshow函数创建一个简单的二维绘图。

## 创建一个二维数组
data = np.random.rand(10, 10)

## 绘制数组
plt.imshow(data)

添加颜色条

现在我们可以使用colorbar函数为绘图添加一个颜色条。

## 为绘图添加一个颜色条
plt.colorbar()

自定义颜色条

我们可以通过使用cmap参数来自定义颜色条的外观。此参数指定用于创建绘图的颜色映射。

## 自定义颜色条
plt.colorbar(cmap='viridis')

给颜色条添加标签

我们可以使用label参数给颜色条添加一个标签。

## 给颜色条添加标签
plt.colorbar(cmap='viridis', label='Random Data')

总结

在本实验中，我们学习了如何向 Matplotlib 绘图添加一个简单的颜色条。我们使用imshow函数创建了一个二维绘图，然后使用colorbar函数添加了一个颜色条。我们还自定义了颜色条的外观并为其添加了一个标签。颜色条是用于可视化绘图中颜色所代表的值范围的有用工具。