使用 Matplotlib 创建视觉上吸引人的柱状图

简介

在本实验中，我们将学习如何使用 Python 的 Matplotlib 库创建柱状图。我们将重点关注如何控制柱子的颜色和图例条目。这将帮助我们创建一个视觉上吸引人且易于阅读和理解的图表。

虚拟机使用提示

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

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

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

导入 Matplotlib 库

首先，我们需要导入 Matplotlib 库。这可以使用以下代码完成：

import matplotlib.pyplot as plt

定义图表数据

接下来，我们需要定义用于创建图表的数据。在这个例子中，我们将创建一个展示不同种类水果供应量的图表。我们将按如下方式定义水果名称、供应数量、条形颜色和图例标签：

fruits = ['apple', 'blueberry', 'cherry', 'orange']
counts = [40, 100, 30, 55]
bar_labels = ['red', 'blue', '_red', 'orange']
bar_colors = ['tab:red', 'tab:blue', 'tab:red', 'tab:orange']

创建柱状图

现在，我们可以使用在步骤 2 中定义的数据来创建柱状图。我们将使用 Matplotlib 的 pyplot 模块的 bar() 方法来创建图表。我们还将分别传入 label 和 color 参数来控制图例条目和条形颜色。以下代码演示了如何创建柱状图：

fig, ax = plt.subplots()
ax.bar(fruits, counts, label=bar_labels, color=bar_colors)
ax.set_ylabel('fruit supply')
ax.set_title('Fruit supply by kind and color')
ax.legend(title='Fruit color')
plt.show()

自定义图表

我们可以通过添加轴标签和标题来进一步自定义图表。我们还可以更改轴标签和图例标题的颜色。以下代码演示了如何自定义图表：

fig, ax = plt.subplots()
ax.bar(fruits, counts, label=bar_labels, color=bar_colors)
ax.set_ylabel('fruit supply', color='blue')
ax.set_xlabel('fruit names', color='blue')
ax.set_title('Fruit supply by kind and color', color='purple')
ax.legend(title='Fruit color', title_color='red', labelcolor='green')
plt.show()

总结

在这个实验中，我们学习了如何使用 Python 的 Matplotlib 库创建柱状图。我们重点关注了如何控制条形的颜色和图例条目。我们还学习了如何通过添加轴标签和标题来自定义图表。按照这些步骤，我们可以创建出视觉上吸引人且易于阅读和理解的图表。