matplotlib: получить макет подзаговора?

Question

У меня есть функция, которая создает сетку похожих 2D гистограмм. Чтобы я мог выбрать, наносить ли этот новый график на уже существующую фигуру, я делаю следующее:

def make_hist2d(x, y, current_fig=False, layout=(1,1,1),*args):

    if current_fig: 

        fig = _plt.gcf()
        ax  = fig.add_subplot(*layout)  # layout=(nrows, ncols, nplot)

    else:

        fig, ax = _plt.subplots()    


    H, x, y = np.histogram2d(...)

    # manipulate the histogram, e.g. column normalize.

    XX, YY = _np.meshgrid(xedges, yedges)
    Image  = ax.pcolormesh(XX, YY, Hplot.T, norm=norm, **pcmesh_kwargs)
    ax.autoscale(tight=True)

    grid_kargs = {'orientation': 'vertical'}
    cax, kw    = _mpl.colorbar.make_axes_gridspec(ax, **grid_kargs)
    cbar       = fig.colorbar(Image, cax=cax)
    cbar.set_label(cbar_title)

    return fig, ax, cbar



def hist2d_grid(data_dict, key_pairs, layout, *args):  # ``*args`` are things like xlog, ylog, xlabel, etc. 
                                                       # that are common to all subplots in the figure.

    fig, ax = _plt.subplots()    

    nplots = range(len(key_pairs) + 1)    # key_pairs = ((k1a, k1b), (k2a, k2b), ..., (kna, knb))

    ax_list = []

    for pair, i in zip(key_pairs, nplots):

        fig, ax, cbar = make_hist2d(data[k1a], data[k1b]

        ax_list.append(ax)

    return fig, ax_list

Тогда я называю что-то вроде:

hgrid = hist2d_grid(...)

Однако, если я хочу добавить новую фигуру в gridЯ не знаю хорошего способа получить макет участка. Например, есть ли что-то вроде:

layout = fig.get_layout()

Это дало бы мне что-то вроде (nrows, ncols, n_subplots)?

Я мог бы сделать это с чем-то вроде:

n_plot = len(ax_list) / 2  # Each subplot generates a plot and a color bar.
n_rows = np.floor(np.sqrt(n_ax))
n_cols = np.ceil(np.sqrt(n_ax))

Но я должен иметь дело с особыми случаями, как (2,4) Массив подзаговоров, за который я бы получил n_rows = 2 а также n_cols = 3, что означает, что я буду проходить (2,3,8) в ax.add_subplot()что явно не работает, потому что 8 > 3*2.

Answer 1

Как ax вернулся fig, ax = plt.subplots(4,2) является массивом осей, то ax.shape предоставит вам информацию о макете, которую вы хотите, например,

 nrows, ncols = ax.shape
 n_subplots = nrows*ncols

Вы также можете получить местоположения различных осей, зацикливая дочерние элементы объекта figure,

[[f.colNum, f.rowNum] for f in fig.get_children()[1:]]

и, вероятно, получить размер от конечного элемента fig.get_children()[-1]

Вы также можете использовать gridspec для более точного определения местоположения участков, если это необходимо. С помощью gridspec вы настраиваете объект gridspec и переходите к субплоту,

import matplotlib.gridspec as gridspec
gs = gridspec.GridSpec(2, 2)
ax = plt.subplot(gs[0, 0])

Чтобы получить макет, который вы можете использовать,

gs.get_geometry()