带箭头的居中脊柱线

Beginner

This tutorial is from open-source community. Access the source code

简介

Matplotlib 是 Python 中一个强大的数据可视化工具。在本教程中,你将学习如何使用 Matplotlib 创建一个带有居中脊柱线和箭头的图表。

虚拟机使用提示

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

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

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

导入必要的库

在创建图表之前,你需要导入必要的库。在这种情况下,你需要 Matplotlib 和 NumPy。

import matplotlib.pyplot as plt
import numpy as np

创建图形和轴对象

接下来,你需要使用 subplots() 函数创建一个图形和轴对象。此函数返回一个 (图形, 轴) 的元组,你可以使用它来修改图表。

fig, ax = plt.subplots()

移动脊柱线

默认情况下,脊柱线绘制在图表的边缘。在这种情况下,你要将左侧和底部的脊柱线移动到图表的中心。

ax.spines[["left", "bottom"]].set_position(("data", 0))

隐藏不必要的脊柱线

你还想隐藏顶部和右侧的脊柱线,因为它们不需要。

ax.spines[["top", "right"]].set_visible(False)

在脊柱线末端绘制箭头

为了指示坐标轴的方向,你可以在脊柱线的末端绘制箭头。

ax.plot(1, 0, ">k", transform=ax.get_yaxis_transform(), clip_on=False)
ax.plot(0, 1, "^k", transform=ax.get_xaxis_transform(), clip_on=False)

向图表中添加数据

最后,你可以向图表中添加一些数据以进行可视化。在这种情况下,你可以使用 plot() 函数来绘制一个正弦波。

x = np.linspace(-0.5, 1., 100)
ax.plot(x, np.sin(x*np.pi))

总结

在本教程中,你学习了如何使用 Matplotlib 创建一个带有居中脊柱线和箭头的图表。你学习了如何将脊柱线移动到图表中心、隐藏不必要的脊柱线以及在脊柱线末端绘制箭头。你还学习了如何向图表中添加数据并进行可视化。