Матплотлиб разных размеров подзаговоров

238

Мне нужно добавить два подзаговора к фигуре. Один участок должен быть примерно в три раза шире второго (такой же высоты). Я справился с этой помощью GridSpecи colspanаргумент , но я хотел бы сделать это с помощью figureтак что я могу сохранить в формате PDF. Я могу откорректировать первую фигуру, используя figsizeаргумент в конструкторе, но как мне изменить размер второго графика?

Джейсон Стримпел
источник
2
Gridspec работает с нормальной фигурой.
Tillsten

Ответы:

372

Другой способ - использовать subplotsфункцию и передать соотношение ширины с помощью 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
источник
1
Спасибо за это; plt.subplotsспособ делать вещи намного чище имо.
Люк Дэвис
2
Я предпочитаю субплоты лучше, чем gridspec, так как вам больше не нужно иметь дело с настройками списка вверх по оси (с gridspec вам все равно нужно создавать ось и графики по одному). Таким образом, сюжеты действительно чище и быстрее в использовании
Eelco van Vliet
3
Что если я хочу, чтобы два участка в одном ряду также различались по высоте? Изменения height_ratioвлияют на всю строку относительно других строк.
Митчелл ван Зуйлен,
Это невозможно с помощью функции subplots. Тем не менее, вы можете добавить что-то вроде этого в приведенный выше код: из mpl_toolkits.axes_grid1 import make_axes_locatable divider = make_axes_locatable (a0) a_empty =divr.append_axes ("bottom", size = "50%") a_empty.axis ('off')
Hagne
2
Я получаю эту ошибку ValueError: Expected the given number of height ratios to match the number of rows of the grid. Я решил это, сказав {'width_ratios':[1]}за 1 строку и т. Д.
Маркус Вебер
223

Вы можете использовать gridspecи 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')

результирующий сюжет

БМУ
источник
31

Вероятно, самый простой способ - использование subplot2grid, описанное в разделе «Настройка местоположения подплота с использованием GridSpec» .

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

равно

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

так пример БМУ становится:

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

Я использовал pyplot«s axesобъект вручную настроить размеры без использования 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
Полезно для тех из нас, кто еще на matplotlib 0.99 без gridspec!
Timday
3
Полезно для тех, где gridspec неадекватен
dreab
Лучший ответ. Как раз то, что нужно, если вы хотите создавать участки разных размеров (2D и 3D) и хотите, чтобы они имели разные размеры. Это также довольно легко понять. Спасибо!
МО