Xlabel / ylabel comum para subparcelas matplotlib

Eu tenho o seguinte enredo:

fig,ax = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

e agora gostaria de fornecer a esse gráfico rótulos comuns do eixo xe do eixo y. Com "comum", quero dizer que deve haver um grande rótulo do eixo x abaixo de toda a grade de subtramas e um grande rótulo do eixo y à direita. Não consigo encontrar nada sobre isso na documentação plt.subplots, e minhas pesquisas sugerem que eu preciso fazer um grande investimento plt.subplot(111)para começar - mas como faço para colocar minhas subparcelas 5 * 2 nisso plt.subplots?

Parece com o que você realmente deseja. Aplica a mesma abordagem desta resposta ao seu caso específico:

import matplotlib.pyplot as plt

fig, ax = plt.subplots(nrows=3, ncols=3, sharex=True, sharey=True, figsize=(6, 6))

fig.text(0.5, 0.04, 'common X', ha='center')
fig.text(0.04, 0.5, 'common Y', va='center', rotation='vertical')

Como considero relevante e elegante o suficiente (não há necessidade de especificar coordenadas para colocar o texto), copio (com uma leve adaptação) uma resposta para outra pergunta relacionada .

import matplotlib.pyplot as plt
fig, axes = plt.subplots(5, 2, sharex=True, sharey=True, figsize=(6,15))
# add a big axis, hide frame
fig.add_subplot(111, frameon=False)
# hide tick and tick label of the big axis
plt.tick_params(labelcolor='none', top=False, bottom=False, left=False, right=False)
plt.xlabel("common X")
plt.ylabel("common Y")

Isso resulta no seguinte (com matplotlib versão 2.2.0):

Sem sharex=True, sharey=Truevocê ter:

Com ele, você deve ficar mais agradável:

fig, axes2d = plt.subplots(nrows=3, ncols=3,
                           sharex=True, sharey=True,
                           figsize=(6,6))

for i, row in enumerate(axes2d):
    for j, cell in enumerate(row):
        cell.imshow(np.random.rand(32,32))

plt.tight_layout()

Mas se você quiser adicionar outros rótulos, adicione-os apenas às plotagens de borda:

fig, axes2d = plt.subplots(nrows=3, ncols=3,
                           sharex=True, sharey=True,
                           figsize=(6,6))

for i, row in enumerate(axes2d):
    for j, cell in enumerate(row):
        cell.imshow(np.random.rand(32,32))
        if i == len(axes2d) - 1:
            cell.set_xlabel("noise column: {0:d}".format(j + 1))
        if j == 0:
            cell.set_ylabel("noise row: {0:d}".format(i + 1))

plt.tight_layout()

A adição de etiquetas para cada plotagem a prejudicaria (talvez haja uma maneira de detectar automaticamente etiquetas repetidas, mas não conheço uma).

Desde o comando:

fig,ax = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

você usou retorna uma tupla composta pela figura e uma lista das instâncias dos eixos, já é suficiente fazer algo como (lembre-se de que mudei fig,axpara fig,axes):

fig,axes = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

for ax in axes:
    ax.set_xlabel('Common x-label')
    ax.set_ylabel('Common y-label')

Se você desejar alterar alguns detalhes de uma subparcela específica, poderá acessá-la através de axes[i]onde iitera suas subparcelas.

Também pode ser muito útil incluir um

fig.tight_layout()

no final do arquivo, antes do plt.show(), para evitar sobreposição de rótulos.

Ficará melhor se você reservar espaço para os rótulos comuns, criando rótulos invisíveis para a subparcela no canto inferior esquerdo. Também é bom passar o tamanho da fonte do rcParams. Dessa forma, os rótulos comuns mudarão de tamanho com sua configuração de rc e os eixos também serão ajustados para deixar espaço para os rótulos comuns.

fig_size = [8, 6]
fig, ax = plt.subplots(5, 2, sharex=True, sharey=True, figsize=fig_size)
# Reserve space for axis labels
ax[-1, 0].set_xlabel('.', color=(0, 0, 0, 0))
ax[-1, 0].set_ylabel('.', color=(0, 0, 0, 0))
# Make common axis labels
fig.text(0.5, 0.04, 'common X', va='center', ha='center', fontsize=rcParams['axes.labelsize'])
fig.text(0.04, 0.5, 'common Y', va='center', ha='center', rotation='vertical', fontsize=rcParams['axes.labelsize'])

Encontrei um problema semelhante ao traçar uma grade de gráficos. Os gráficos consistiam em duas partes (superior e inferior). O rótulo y deveria estar centrado nas duas partes.

Eu não queria usar uma solução que dependesse de conhecer a posição na figura externa (como fig.text ()), então manipulei a posição y da função set_ylabel (). Geralmente é 0,5, no meio do gráfico ao qual é adicionado. Como o preenchimento entre as partes (hspace) no meu código era zero, eu pude calcular o meio das duas partes em relação à parte superior.

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

# Create outer and inner grid
outerGrid = gridspec.GridSpec(2, 3, width_ratios=[1,1,1], height_ratios=[1,1])
somePlot = gridspec.GridSpecFromSubplotSpec(2, 1,
               subplot_spec=outerGrid[3], height_ratios=[1,3], hspace = 0)

# Add two partial plots
partA = plt.subplot(somePlot[0])
partB = plt.subplot(somePlot[1])

# No x-ticks for the upper plot
plt.setp(partA.get_xticklabels(), visible=False)

# The center is (height(top)-height(bottom))/(2*height(top))
# Simplified to 0.5 - height(bottom)/(2*height(top))
mid = 0.5-somePlot.get_height_ratios()[1]/(2.*somePlot.get_height_ratios()[0])
# Place the y-label
partA.set_ylabel('shared label', y = mid)

plt.show()

cenário

Desvantagens:

• A distância horizontal ao gráfico é baseada na parte superior, os marcadores inferiores podem se estender para o rótulo.

• A fórmula não leva em consideração o espaço entre as partes.

• Lança uma exceção quando a altura da parte superior é 0.

Provavelmente existe uma solução geral que leva em consideração o preenchimento entre os números.

Atualizar:

Esse recurso agora faz parte do pacote proplot matplotlib que eu liberei recentemente no pypi. Por padrão, quando você cria figuras, os rótulos são "compartilhados" entre os eixos.

Resposta original:

Eu descobri um método mais robusto:

Se você conhece o bottome topkwargs que foram GridSpeciniciados, ou conhece as posições das arestas dos seus eixos em Figurecoordenadas , também pode especificar a posição do marcador em Figurecoordenadas com alguma mágica "transformada" sofisticada. Por exemplo:

import matplotlib.transforms as mtransforms
bottom, top = .1, .9
f, a = plt.subplots(nrows=2, ncols=1, bottom=bottom, top=top)
avepos = (bottom+top)/2
a[0].yaxis.label.set_transform(mtransforms.blended_transform_factory(
       mtransforms.IdentityTransform(), f.transFigure # specify x, y transform
       )) # changed from default blend (IdentityTransform(), a[0].transAxes)
a[0].yaxis.label.set_position((0, avepos))
a[0].set_ylabel('Hello, world!')

... e você deve ver que o rótulo ainda se ajusta de maneira adequada à esquerda e à direita para não se sobrepor aos ticklabels, como normalmente - mas agora ele será ajustado para estar sempre exatamente entre as subparcelas desejadas.

Além disso, se você nem usar set_position, o rótulo será exibido por padrão exatamente na metade da figura . Suponho que isso ocorre porque, quando o rótulo é finalmente desenhado, matplotlibusa 0,5 para o y-coordinate sem verificar se a transformação de coordenadas subjacente foi alterada.