将现有的 matplotlib 数字放入子图中

Question

我想要一些关于如何安排 matplotlib.figure.Figure 对象的建议

I make an object of type 'matplotlib.figure.Figure' using the following function ( https://nilearn.github.io/modules/generated/nilearn.plotting.plot_surf_roi.html ):

from nilearn import plotting

def plot_surf(surface_data, view, fig):
    img = plotting.plot_surf_roi(surface_data['surf_mesh'], 
                        roi_map=surface_data['comp_labels'],
                        hemi=hemi, view=view,
                        cmap='RdBu_r',
                        vmax=np.nanmax(surface_data['comp_labels']), 
                        vmin=np.nanmin(surface_data['comp_labels']),
                        bg_map=surface_data['bg_maps'],
                        darkness=0.6,
                        bg_on_data=True,
                        title='',
                        figure = fig)
    return img

我想制作其中的一些并安排为子图。 到目前为止，这是我不成功的尝试：

fig = plt.figure()
fig, ax = plt.subplots()
plot_surf(surface_data, 'lateral', fig)
plot_surf(surface_data, 'medial', fig)
plt.show()

任何建议表示赞赏

Answer 1

尝试这样做：

def plot_surf(surface_data, view, fig, axes):
    img = plotting.plot_surf_roi(surface_data['surf_mesh'], 
                        roi_map=surface_data['comp_labels'],
                        hemi=hemi, view=view,
                        cmap='RdBu_r',
                        vmax=np.nanmax(surface_data['comp_labels']), 
                        vmin=np.nanmin(surface_data['comp_labels']),
                        bg_map=surface_data['bg_maps'],
                        darkness=0.6,
                        bg_on_data=True,
                        title='',
                        figure = fig,
                        axes=axes)
    return img

fig, (ax1, ax2) = plt.subplots(2, subplot_kw={'projection': '3d'}) 
plot_surf(surface_data, 'lateral', fig, ax1)
plot_surf(surface_data, 'medial', fig, ax2)