matplotlib给坐标轴两端加上箭头
可以使用matplotlib中的arrow函数来给坐标轴两端加上箭头。具体实现方法如下:
- 导入matplotlib库
import matplotlib.pyplot as plt
- 创建一个图形对象和一个子图对象
fig, ax = plt.subplots()
- 绘制图形
# 绘制函数曲线或散点图等
- 给x轴和y轴加上箭头
# 给x轴加上箭头
ax.annotate('', xy=(max_x, 0), xytext=(min_x, 0), arrowprops=dict(arrowstyle='->'))
# 给y轴加上箭头
ax.annotate('', xy=(0, max_y), xytext=(0, min_y), arrowprops=dict(arrowstyle='->'))
其中,arrow函数的参数arrowstyle可以设置箭头的样式,如'->'表示带箭头的直线。
完整代码示例:
import matplotlib.pyplot as plt
import numpy as np
# 创建数据
x = np.linspace(-5, 5, 100)
y = np.sin(x)
# 创建图形对象和一个子图对象
fig, ax = plt.subplots()
# 绘制函数曲线
ax.plot(x, y)
# 获取x轴和y轴的最大值和最小值
min_x, max_x = ax.get_xlim()
min_y, max_y = ax.get_ylim()
# 给x轴加上箭头
ax.annotate('', xy=(max_x, 0), xytext=(min_x, 0), arrowprops=dict(arrowstyle='->'))
# 给y轴加上箭头
ax.annotate('', xy=(0, max_y), xytext=(0, min_y), arrowprops=dict(arrowstyle='->'))
# 显示图形
plt.show()
原文地址: https://www.cveoy.top/t/topic/bxur 著作权归作者所有。请勿转载和采集!