Matplotlib subplots kích thước khác nhau

Tôi cần thêm hai ô con vào một hình. Một subplot cần rộng gấp khoảng ba lần so với cái thứ hai (cùng chiều cao). Tôi đã hoàn thành việc này bằng cách sử dụng GridSpecvà colspanđối số nhưng tôi muốn làm điều này bằng cách sử dụng figuređể tôi có thể lưu vào PDF. Tôi có thể điều chỉnh hình đầu tiên bằng cách sử dụng figsizeđối số trong hàm tạo, nhưng làm cách nào để thay đổi kích thước của ô thứ hai?

Một cách khác là sử dụng subplotshàm và vượt qua tỷ lệ chiều rộng với 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')

Bạn có thể sử dụng gridspecvà 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')

Có lẽ cách đơn giản nhất là sử dụng subplot2grid, được mô tả trong Tùy chỉnh vị trí của Subplot bằng GridSpec .

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

bằng

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

vì vậy ví dụ của bmu trở thành:

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

Tôi đã từng pyplot's axesđối tượng để tự điều chỉnh kích thước mà không sử dụng 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()