如何在Matplotlib中强制显示正方形网格单元？

嘿，我完全懂你的需求——你要的不是把整个绘图窗口变成正方形，而是不管图的整体宽高比例、轴的显示范围怎么变，每个网格单元都得是规规矩矩的正方形，对吧？我之前在Jupyter Notebook里也踩过axis('equal')的坑，后来摸索出了更靠谱的解决办法。

为什么axis('equal')可能失效？ #

plt.axis('equal')确实是用来让x/y轴单位长度视觉相等的，但有时候在Jupyter的inline绘图模式下，或者当你已经设置了画布尺寸（figsize）、轴范围后，它的效果可能被覆盖，或者因为调用时机不对而没生效。更可靠的方式是直接针对轴对象（Axes）进行设置。

正确的实现方法 #

核心就是用ax.set_aspect('equal')，这个方法能强制让x轴和y轴上的每个数据单位在视觉上长度一致，不管你的画布是宽屏还是竖屏，也不管轴的显示范围多大，网格单元都会保持正方形。

下面是一个完整的2D向量空间绘图示例：

import matplotlib.pyplot as plt
import numpy as np

# 1. 创建画布和轴对象，故意用非正方形的尺寸测试
fig, ax = plt.subplots(figsize=(8, 5))  # 宽8高5的画布，不是正方形

# 2. 绘制示例向量
v1 = np.array([2, 3])
v2 = np.array([-1, 4])
ax.quiver(0, 0, v1[0], v1[1], angles='xy', scale_units='xy', scale=1, color='#ff4444')
ax.quiver(0, 0, v2[0], v2[1], angles='xy', scale_units='xy', scale=1, color='#33b5e5')

# 3. 设置轴范围（可以根据你的需求调整）
ax.set_xlim(-2, 3)
ax.set_ylim(0, 5)

# 4. 关键：强制轴的纵横比相等，保证网格单元正方形
ax.set_aspect('equal')

# 5. 添加网格和标注
ax.grid(True, which='both', linestyle='--', alpha=0.6)
ax.set_xlabel('X轴')
ax.set_ylabel('Y轴')
ax.text(v1[0]+0.1, v1[1], '$\mathbf{v_1}$', color='#ff4444', fontsize=12)
ax.text(v2[0]+0.1, v2[1], '$\mathbf{v_2}$', color='#33b5e5', fontsize=12)

plt.show()

额外说明 #

• 如果你是用plt.plot等其他绘图函数，同样可以用plt.gca().set_aspect('equal')来获取当前轴对象并设置纵横比，比如：

plt.figure(figsize=(10, 4))
plt.plot([0, 3], [0, 2])
plt.grid(True)
plt.gca().set_aspect('equal')  # 获取当前轴并设置
plt.show()

• 这个方法不受画布尺寸、轴范围的影响，哪怕你把画布拉得很宽或者很窄，网格里的每个小格子都会是正方形，x轴上的1个单位和y轴上的1个单位视觉长度完全一致。

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