Matplotlib subplots ขนาดแตกต่างกัน


237

ฉันต้องเพิ่มสองแผนการย่อยลงในรูป หนึ่งแผนย่อยจะต้องมีความกว้างประมาณสามเท่าของที่สอง (ความสูงเดียวกัน) ฉันใช้วิธีนี้GridSpecกับการcolspanโต้แย้งสำเร็จแล้วแต่ฉันต้องการทำสิ่งนี้โดยใช้figureดังนั้นฉันสามารถบันทึกเป็น PDF ได้ ฉันสามารถปรับรูปแรกโดยใช้figsizeอาร์กิวเมนต์ในตัวสร้าง แต่ฉันจะเปลี่ยนขนาดของพล็อตที่สองได้อย่างไร


2
Gridspec ทำงานร่วมกับตัวเลขปกติ
จนถึง

คำตอบ:


371

อีกวิธีคือใช้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')

1
ขอบคุณสำหรับสิ่งนี้; plt.subplotsวิธีการทำสิ่งที่มากสะอาด IMO
ลุคเดวิส

2
ฉันชอบแผนย่อยที่ดีกว่ากริดสเปคเพราะคุณไม่ต้องจัดการกับการตั้งค่ารายการของแกนอีกต่อไป (กับกริดสเปคคุณยังต้องทำให้แกนและพล็อตทีละหนึ่ง) ดังนั้นแผนการย่อยจึงสะอาดและใช้งานได้เร็วขึ้น
Eelco van Vliet

3
ถ้าฉันต้องการให้แปลงสองผืนในหนึ่งแถวมีความสูงแตกต่างกันด้วยล่ะ การเปลี่ยนแปลงheight_ratioปรากฏเพื่อส่งผลกระทบต่อทั้งแถวเทียบกับแถวอื่น ๆ
Mitchell van Zuylen

ไม่สามารถทำได้ผ่านฟังก์ชั่นย่อย อย่างไรก็ตามคุณสามารถเพิ่มสิ่งนี้ลงในโค้ดด้านบน: จาก mpl_toolkits.axes_grid1 อิมปอร์ต make_axes_locatable divider = make_axes_locatable (a0) a_empty = divider.append_axes ("ด้านล่าง", ขนาด = "50%") a_empty.axis
Hagne

2
ฉันได้รับข้อผิดพลาดValueError: Expected the given number of height ratios to match the number of rows of the gridนี้ ฉันแก้ไขมันด้วยการพูด{'width_ratios':[1]}1 แถว ฯลฯ
Markus Weber

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

ดังนั้นตัวอย่างของ 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')

29

ผมใช้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()

2
มีประโยชน์สำหรับพวกเราที่ยังใช้งาน matplotlib 0.99 โดยไม่มีกริดสเปค!
timday

3
มีประโยชน์สำหรับผู้ที่ gridspec ไม่เพียงพอ
dreab

คำตอบที่ดีที่สุด สิ่งที่จำเป็นถ้าคุณต้องการทำ subplots ที่มีขนาดต่างกัน (2D และ 3D) และต้องการให้มันมีขนาดแตกต่างกัน นอกจากนี้ยังเข้าใจง่าย ขอบคุณ!
MO
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.