Matplotlib różne wykresy wielkości

237

Muszę dodać dwie figury do figury. Jeden wykres podrzędny musi być około trzy razy szerszy niż drugi (ta sama wysokość). I dokonał tego przy użyciu GridSpeci colspanargumentu, ale chciałbym zrobić to za pomocą figuretak mogę zapisać do pliku PDF. Mogę dostosować pierwszą cyfrę za pomocą figsizeargumentu w konstruktorze, ale jak zmienić rozmiar drugiego wykresu?

Jason Strimpel
źródło
2
Gridspec działa z normalną postacią.
tillsten

Odpowiedzi:

371

Innym sposobem jest użycie subplotsfunkcji i przekazanie współczynnika szerokości z 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
źródło
1
Dzięki za to; plt.subplotssposób robienia rzeczy jest znacznie czystsze imo.
Luke Davis
2
Lubię wykresy podrzędne lepiej niż specyfikacja siatki, ponieważ nie musisz już zajmować się ustawieniami listy dla osi (w przypadku specyfikacji siatki nadal musisz tworzyć oś i wykresy jeden po drugim). Tak więc wątki podrzędne są w rzeczywistości czystsze i szybsze w użyciu
Eelco van Vliet
3
Co jeśli chcę, aby dwie działki w jednym rzędzie również różniły się wysokością? height_ratioWydaje się, że zmiana wpływa na cały wiersz w stosunku do innych wierszy.
Mitchell van Zuylen
Nie jest to możliwe dzięki funkcji podplotów. Możesz jednak dodać coś takiego do powyższego kodu: z 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
Otrzymuję ten błąd ValueError: Expected the given number of height ratios to match the number of rows of the grid. Rozwiązałem to, mówiąc {'width_ratios':[1]}o 1 rzędzie itp.
Markus Weber
223

Możesz użyć gridspeci 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')

wynikowa fabuła

bmu
źródło
31

Prawdopodobnie jest to najprostszy sposób subplot2gridopisany w Dostosowywanie położenia podplotu za pomocą GridSpec .

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

jest równe

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

przykład bmu brzmi:

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')
endolit
źródło
29

Że stosuje się pyplot„a axesobiekt do ręcznej regulacji wielkości bez użycia 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
źródło
2
Przydatne dla tych z nas, którzy wciąż korzystają z Matplotlib 0.99 bez specyfikacji grid!
dnia
3
Przydatne dla tych, w których specyfikacja siatki jest nieodpowiednia
dreab
Najlepsza odpowiedź w historii. Dokładnie to, co jest potrzebne, jeśli chcesz wykonać wykresy podrzędne o różnych wymiarach (2D i 3D) i chcesz, aby miały one różne rozmiary. Jest to również dość łatwe do zrozumienia. Dziękuję Ci!
MO