matplotlib:如何返回matplotlib对象，然后作为subplot绘制？

import matplotlib.pyplot as plt

fig = plt.figure()

ax1 = fig.add_subplot(2, 2, 1)  # 创建一个2x2的subplot，并选择第1个位置
ax2 = fig.add_subplot(2, 2, 2)  # 创建一个2x2的subplot，并选择第2个位置
ax3 = fig.add_subplot(2, 2, 3)  # 创建一个2x2的subplot，并选择第3个位置
ax4 = fig.add_subplot(2, 2, 4)  # 创建一个2x2的subplot，并选择第4个位置

ax1.plot([1, 2, 3, 4], [1, 4, 2, 3])  # 在第1个subplot上绘制线图
ax2.scatter([1, 2, 3, 4], [1, 4, 2, 3])  # 在第2个subplot上绘制散点图
ax3.bar([1, 2, 3, 4], [1, 4, 2, 3])  # 在第3个subplot上绘制柱状图
ax4.pie([1, 2, 3, 4], labels=['A', 'B', 'C', 'D'])  # 在第4个subplot上绘制饼图

plt.show()