在matplotlib中将轴投影从二维更改为三维

在matplotlib中创建轴后，是否可以更改轴的投影

我正在尝试编写用于绘图的通用代码，使每个绘图都成为一个函数，该函数将轴作为参数来处理

例如

但是，对于三维绘图（例如，使用绘图曲面），需要一个不同于标准二维轴的三维投影（例如，通过ax=fig.add_子图（111，projection='3d'）初始化）

如果给我的函数一个带有2D投影的ax，我如何将其更改为3D

谢谢

def plot_this(ax):
    ax.plot((1,2,3))

def plot_that(ax):
    ax.plot_surface(etc.)