简介
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 创建一个带有居中脊柱线和箭头的图表。你学习了如何将脊柱线移动到图表中心、隐藏不必要的脊柱线以及在脊柱线末端绘制箭头。你还学习了如何向图表中添加数据并进行可视化。