Matplotlib Quiver图未适配箭头范围，如何自动调整坐标轴？

使用matplotlib.pyplot的quiver绘制箭头时，箭头常超出图像范围。图表仅适配箭头起点（即quiver()的X、Y参数），未考虑箭头实际延伸范围。想知道是否有类似tight layout的全局命令，可一次性将所有箭头纳入画布可见区域（支持多图表场景）？已知可以用plt.xlim()、plt.ylim()或Axes.set_xlim/Axes.set_ylim方法，但希望有更简便的全局方式。

补充测试示例 #

尝试了constrained layout，但无效：

未启用constrained layout的代码 #

import matplotlib.pyplot as plt
import numpy as np

pts = np.array([[1, 2], [3, 4]])
end_pts = np.array([[2, 4], [6, 8]])
diff = end_pts - pts

plt.quiver(pts[:,0], pts[:,1], diff[:,0], diff[:,1],
           angles='xy', scale_units='xy', scale=1.)

运行后有箭头超出画布。

启用constrained layout的代码 #

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams['figure.constrained_layout.use'] = True

pts = np.array([[1, 2], [3, 4]])
end_pts = np.array([[2, 4], [6, 8]])
diff = end_pts - pts

plt.quiver(pts[:,0], pts[:,1], diff[:,0], diff[:,1],
           angles='xy', scale_units='xy', scale=1.)

结果仅缩小边距，坐标轴范围无变化。

解决方案 #

目前Matplotlib没有内置的全局命令自动适配quiver的箭头范围，不过可以通过以下两种方式实现：

方法1：手动计算极值并设置坐标轴范围 #

合并所有箭头的起点和终点坐标，计算极值后给坐标轴加一点余量，确保箭头完整显示：

import matplotlib.pyplot as plt
import numpy as np

pts = np.array([[1, 2], [3, 4]])
end_pts = np.array([[2, 4], [6, 8]])
diff = end_pts - pts

plt.quiver(pts[:,0], pts[:,1], diff[:,0], diff[:,1],
           angles='xy', scale_units='xy', scale=1.)

# 收集所有起点和终点的x、y坐标
all_x = np.concatenate([pts[:,0], end_pts[:,0]])
all_y = np.concatenate([pts[:,1], end_pts[:,1]])

# 预留10%的余量，避免箭头贴边
margin = 0.1
plt.xlim(all_x.min() - margin, all_x.max() + margin)
plt.ylim(all_y.min() - margin, all_y.max() + margin)

plt.show()

方法2：自定义函数实现多子图自动适配 #

如果需要处理多子图场景，可以写一个函数遍历每个子图的quiver元素，自动计算并调整坐标轴范围：

import matplotlib.pyplot as plt
import numpy as np

def adjust_quiver_limits(ax=None, margin=0.1):
    if ax is None:
        ax = plt.gca()  # 获取当前活动坐标轴
    all_x, all_y = [], []
    # 遍历坐标轴上的所有quiver对象
    for artist in ax.get_children():
        if isinstance(artist, plt.quiver.Quiver):
            # 获取箭头起点坐标
            x, y = artist.get_offsets().T
            # 获取向量分量，计算终点坐标
            u, v = artist.get_UVC()
            end_x = x + u
            end_y = y + v
            # 收集所有坐标点
            all_x.extend(x.tolist() + end_x.tolist())
            all_y.extend(y.tolist() + end_y.tolist())
    if all_x and all_y:
        # 调整坐标轴范围
        ax.set_xlim(min(all_x) - margin, max(all_x) + margin)
        ax.set_ylim(min(all_y) - margin, max(all_y) + margin)

# 测试多子图场景
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(10, 4))

# 子图1绘制箭头
pts1 = np.array([[1, 2], [3, 4]])
end_pts1 = np.array([[2, 4], [6, 8]])
diff1 = end_pts1 - pts1
ax1.quiver(pts1[:,0], pts1[:,1], diff1[:,0], diff1[:,1], angles='xy', scale_units='xy', scale=1.)
ax1.set_title("子图1")

# 子图2绘制箭头
pts2 = np.array([[0, 0], [5, 5]])
end_pts2 = np.array([[3, 6], [10, 0]])
diff2 = end_pts2 - pts2
ax2.quiver(pts2[:,0], pts2[:,1], diff2[:,0], diff2[:,1], angles='xy', scale_units='xy', scale=1.)
ax2.set_title("子图2")

# 自动调整两个子图的坐标轴范围
adjust_quiver_limits(ax1)
adjust_quiver_limits(ax2)

plt.tight_layout()
plt.show()

内容的提问来源于stack exchange，提问作者dasWesen