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