Subplots de tamanhos diferentes do Matplotlib

237

Eu preciso adicionar duas subtramas a uma figura. Uma subparcela precisa ter cerca de três vezes a largura da segunda (mesma altura). Eu realizei esse uso GridSpece o colspanargumento, mas gostaria de fazer isso usando figurepara que eu possa salvar em PDF. Posso ajustar a primeira figura usando o figsizeargumento do construtor, mas como altero o tamanho do segundo gráfico?

Jason Strimpel
fonte
2
Gridspec trabalha com uma figura normal.
tillsten

Respostas:

371

Outra maneira é usar a subplotsfunção e passar a taxa de largura com 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')
Hagne
fonte
1
Obrigado por isso; a plt.subplotsmaneira de fazer as coisas é muito mais limpa.
Luke Davis
2
Eu gosto mais de subparcelas que gridspec, pois você não precisa mais lidar com as configurações da lista para o eixo (com gridspec você ainda precisa fazer o eixo e os gráficos um por um). Então subtramas é mais limpo e mais rápido para utilização na verdade
Eelco van Vliet
3
E se eu quiser que as duas parcelas em uma linha também diferem em altura? A alteração height_ratioparece afetar a linha inteira em relação a outras linhas.
Mitchell van Zuylen
Isso não é possível através da função subplots. No entanto, você pode adicionar algo assim ao código acima: de 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
Estou recebendo esse erro ValueError: Expected the given number of height ratios to match the number of rows of the grid. Eu resolvi dizendo {'width_ratios':[1]}1 linha, etc. #
Markus Weber
223

Você pode usar 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')

plot resultante

bmu
fonte
31

Provavelmente, a maneira mais simples de usar é subplot2griddescrita em Personalizando o local da subtrama usando o GridSpec .

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

é igual a

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

então o exemplo de bmu se torna:

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')
endólito
fonte
29

Usei pyploto axesobjeto para ajustar manualmente os tamanhos sem usar 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()
Jason Strimpel
fonte
2
Útil para aqueles de nós ainda no matplotlib 0.99 sem gridspec!
timday 18/08/12
3
Útil para aqueles onde gridspec é inadequada
dreab
Melhor resposta de sempre. Exatamente o que é necessário se você deseja fazer subparcelas de diferentes dimensões (2D e 3D) e desejar que eles tenham tamanhos diferentes. Também é muito fácil de entender. Obrigado!
MO