Подзаголовки разного размера в matplotlib

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


person Jason Strimpel    schedule 30.04.2012    source источник


Ответы (5)


Другой способ - использовать функцию 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')

введите описание изображения здесь

person Hagne    schedule 09.03.2016
comment
Для тех, кому интересно: height_ratios можно использовать для относительного масштабирования подзаголовка высот, docs здесь. - person blsqr; 30.06.2021
comment
Также см. Руководство по gridspec и gridspec_kw здесь. - person spinup; 08.07.2021

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

введите описание изображения здесь

person bmu    schedule 02.05.2012

Я использовал объект pyplot 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()

введите описание изображения здесь

person Jason Strimpel    schedule 01.05.2012

Вероятно, самый простой способ - использовать 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])

так что пример bmu становится:

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')

введите описание изображения здесь

person endolith    schedule 08.04.2013

Проще говоря, подзагрузка другого размера может выполняться без gridspec:

plt.figure(figsize=(12, 6))
ax1 = plt.subplot(2,3,1)
ax2 = plt.subplot(2,3,2)
ax3 = plt.subplot(2,3,3)
ax4 = plt.subplot(2,1,2)
axes = [ax1, ax2, ax3, ax4]

Вывод

person Sohail Ahmed    schedule 10.04.2021
comment
plt.subplot объяснение: subplot(nrows, ncols, index, **kwargs) - person Seth; 20.07.2021