Передайте оси в функцию и создайте подзаголовки

#python #plot

Вопрос:

Я написал несколько функций построения графиков со многими входными параметрами:

 lineplot_custom()
barchart_custom()
heatmap_custom()
...
 

Каждый из них возвращает matplotlib.axes._subplots.AxesSubplot объект. Сейчас я пытаюсь написать функцию, которая позволит мне объединить некоторые matplotlib.axes._subplots.AxesSubplot объекты и отобразить их вместе в подзаголовке.

Я хотел бы иметь возможность заставить что-то подобное работать:

 created_axs = [AxesSubplot1, AxesSubplot2, AxesSubplot3, AxesSubplot4]

def create_subplot(created_axs)
    fig, ((ax1, ax2), (ax3, ax4)) = plt.subplots(2,2)
    ax1 = created_axs[0]
    ax2 = created_axs[1]
    ax3 = created_axs[2]
    ax2 = created_axs[3]
    return fig
 

Я также пробовал использовать эту функцию axes.append :

 fig = plt.figure(figsize=(20,10))
axs = []

for i in range(4):
    plt.subplot(2,2,i 1)
    axs.append(created_axs[i])