Matplotlib - 如何在图像上的不同轴上叠加网格？

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

# 创建一个2x2的GridSpec
gs = gridspec.GridSpec(2, 2)

# 创建一个figure
fig = plt.figure()

# 在第一个子图上绘制数据
ax1 = fig.add_subplot(gs[0, 0])
ax1.plot([0, 1], [0, 1])
ax1.set_title('Ax1')

# 在第二个子图上绘制数据
ax2 = fig.add_subplot(gs[0, 1], sharey=ax1)
ax2.plot([0, 1], [1, 0])
ax2.set_title('Ax2')
ax2.set_yticks([])

# 在第三个子图上绘制数据
ax3 = fig.add_subplot(gs[1, :], sharex=ax1)
ax3.plot([0, 1], [0, 1])
ax3.set_title('Ax3')
ax3.set_xticks([])

# 在所有子图上添加网格线
for ax in fig.axes:
    ax.grid(True)

# 显示图像
plt.show()