python - seleccion - Encabezados de fila y columna en las subparcelas de matplotlib

seleccionar columnas de un dataframe pandas

Question

¿Cuál es la mejor práctica para agregar una fila y un encabezado de columna a una grilla de subparcelas generadas en un bucle en matplotlib ? Puedo pensar en una pareja, pero no especialmente clara:

1. Para las columnas, con un contador para su ciclo, puede usar set_title() para la primera fila. Para las filas esto no funciona. Tendría que dibujar text fuera de las tramas.
2. Agregue una fila adicional de subtramas en la parte superior y una columna adicional de subtramas en la izquierda, y dibuje texto en el medio de esa subtrama.

¿Puedes sugerir una mejor alternativa?

Answer 1

Hay varias formas de hacer esto. La manera fácil es explotar las etiquetas y los títulos de la gráfica y luego usar fig.tight_layout() para dejar espacio para las etiquetas. Alternativamente, puede colocar texto adicional en la ubicación correcta con annotate y luego dejar espacio para ello de forma semi-manual.

Si no tiene etiquetas y en sus ejes, es fácil explotar el título y la etiqueta y de la primera fila y columna de ejes.

import matplotlib.pyplot as plt

cols = [''Column {}''.format(col) for col in range(1, 4)]
rows = [''Row {}''.format(row) for row in [''A'', ''B'', ''C'', ''D'']]

fig, axes = plt.subplots(nrows=4, ncols=3, figsize=(12, 8))

for ax, col in zip(axes[0], cols):
    ax.set_title(col)

for ax, row in zip(axes[:,0], rows):
    ax.set_ylabel(row, rotation=0, size=''large'')

fig.tight_layout()
plt.show()

Si tiene etiquetas y, o si prefiere un poco más de flexibilidad, puede usar annotate para colocar las etiquetas. Esto es más complicado, pero le permite tener títulos de parcelas individuales, etiquetas y demás, además de las etiquetas de fila y columna.

import matplotlib.pyplot as plt
from matplotlib.transforms import offset_copy


cols = [''Column {}''.format(col) for col in range(1, 4)]
rows = [''Row {}''.format(row) for row in [''A'', ''B'', ''C'', ''D'']]

fig, axes = plt.subplots(nrows=4, ncols=3, figsize=(12, 8))
plt.setp(axes.flat, xlabel=''X-label'', ylabel=''Y-label'')

pad = 5 # in points

for ax, col in zip(axes[0], cols):
    ax.annotate(col, xy=(0.5, 1), xytext=(0, pad),
                xycoords=''axes fraction'', textcoords=''offset points'',
                size=''large'', ha=''center'', va=''baseline'')

for ax, row in zip(axes[:,0], rows):
    ax.annotate(row, xy=(0, 0.5), xytext=(-ax.yaxis.labelpad - pad, 0),
                xycoords=ax.yaxis.label, textcoords=''offset points'',
                size=''large'', ha=''right'', va=''center'')

fig.tight_layout()
# tight_layout doesn''t take these labels into account. We''ll need 
# to make some room. These numbers are are manually tweaked. 
# You could automatically calculate them, but it''s a pain.
fig.subplots_adjust(left=0.15, top=0.95)

plt.show()

Answer 2

La respuesta anterior funciona. Solo que no en la segunda versión de la respuesta, tienes:

for ax, row in zip(axes[:,0], rows):
    ax.annotate(col, xy=(0, 0.5), xytext=(-ax.yaxis.labelpad-pad,0),
                xycoords=ax.yaxis.label, textcoords=''offset points'',
                size=''large'', ha=''right'', va=''center'')

en lugar de:

for ax, row in zip(axes[:,0], rows):
    ax.annotate(row,xy=(0, 0.5), xytext=(-ax.yaxis.labelpad-pad,0),                    
                xycoords=ax.yaxis.label, textcoords=''offset points'',
                size=''large'', ha=''right'', va=''center'')