Matplotlib sottotrame di dimensioni diverse


237

Devo aggiungere due sottotrame a una figura. Una sottotrama deve essere larga circa tre volte la seconda (stessa altezza). Ho realizzato questo usandoGridSpec e l' colspanargomento ma mi piacerebbe farlo usando figurecosì posso salvare in PDF. Posso regolare la prima figura usando l' figsizeargomento nel costruttore, ma come posso cambiare la dimensione del secondo grafico?


2
Gridspec funziona con una figura normale.
tillsten,

Risposte:


371

Un altro modo è utilizzare la subplotsfunzione e passare il rapporto larghezza con gridspec_kw:

import numpy as np
import matplotlib.pyplot as plt 

# generate some data
x = np.arange(0, 10, 0.2)
y = np.sin(x)

# plot it
f, (a0, a1) = plt.subplots(1, 2, gridspec_kw={'width_ratios': [3, 1]})
a0.plot(x, y)
a1.plot(y, x)

f.tight_layout()
f.savefig('grid_figure.pdf')

1
Grazie per questo; il plt.subplotsmodo di fare le cose è molto più pulito.
Luke Davis,

2
Mi piacciono le trame secondarie meglio di gridspec in quanto non è più necessario gestire le impostazioni nell'elenco per l'asse (con gridspec è comunque necessario creare l'asse e i grafici uno per uno). Quindi le sottotrame sono più pulite e più veloci da usare davvero
Eelco van Vliet

3
Cosa succede se voglio che i due grafici in una riga differiscano anche in altezza? La modifica height_ratiosembra influire sull'intera riga rispetto ad altre righe.
Mitchell van Zuylen,

Questo non è possibile tramite la funzione di sottotrame. Tuttavia, potresti aggiungere qualcosa del genere al codice sopra: da mpl_toolkits.axes_grid1 import make_axes_locatable divider = make_axes_locatable (a0) a_empty = divider.append_axes ("bottom", size = "50%") a_empty.axis ('off')
Hagne,

2
Ricevo questo errore ValueError: Expected the given number of height ratios to match the number of rows of the grid. L'ho risolto dicendo {'width_ratios':[1]}per 1 riga, ecc.
Markus Weber il

223

Puoi usare gridspece figure:

import numpy as np
import matplotlib.pyplot as plt 
from matplotlib import gridspec

# generate some data
x = np.arange(0, 10, 0.2)
y = np.sin(x)

# plot it
fig = plt.figure(figsize=(8, 6)) 
gs = gridspec.GridSpec(1, 2, width_ratios=[3, 1]) 
ax0 = plt.subplot(gs[0])
ax0.plot(x, y)
ax1 = plt.subplot(gs[1])
ax1.plot(y, x)

plt.tight_layout()
plt.savefig('grid_figure.pdf')

trama risultante


31

Probabilmente si sta utilizzando il modo più semplice subplot2grid, descritto in Personalizzazione della posizione della sottotrama mediante GridSpec .

ax = plt.subplot2grid((2, 2), (0, 0))

è uguale a

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

così l'esempio di bmu diventa:

import numpy as np
import matplotlib.pyplot as plt

# generate some data
x = np.arange(0, 10, 0.2)
y = np.sin(x)

# plot it
fig = plt.figure(figsize=(8, 6))
ax0 = plt.subplot2grid((1, 3), (0, 0), colspan=2)
ax0.plot(x, y)
ax1 = plt.subplot2grid((1, 3), (0, 2))
ax1.plot(y, x)

plt.tight_layout()
plt.savefig('grid_figure.pdf')

29

Ho usato pyplotl' axesoggetto per regolare manualmente le dimensioni senza usare GridSpec:

import matplotlib.pyplot as plt
import numpy as np
x = np.arange(0, 10, 0.2)
y = np.sin(x)

# definitions for the axes
left, width = 0.07, 0.65
bottom, height = 0.1, .8
bottom_h = left_h = left+width+0.02

rect_cones = [left, bottom, width, height]
rect_box = [left_h, bottom, 0.17, height]

fig = plt.figure()

cones = plt.axes(rect_cones)
box = plt.axes(rect_box)

cones.plot(x, y)

box.plot(y, x)

plt.show()

2
Utile per quelli di noi che sono ancora su matplotlib 0.99 senza gridspec!
martedì

3
Utile per coloro in cui gridspec è inadeguato
brutto

La migliore risposta di sempre. Esattamente ciò che è necessario se si desidera eseguire sottotrame di dimensioni diverse (2D e 3D) e si desidera che abbiano dimensioni diverse. È anche abbastanza facile da capire. Grazie!
Lunedì
Utilizzando il nostro sito, riconosci di aver letto e compreso le nostre Informativa sui cookie e Informativa sulla privacy.
Licensed under cc by-sa 3.0 with attribution required.