Matplotlib不同大小的子情节

Question

我需要在一个图形中添加两个子图。一个子图的宽度必须是第二个子图的三倍（相同高度）。我使用GridSpec和colspan参数实现了这一点，但我想使用figure实现这一点，这样我就可以保存到PDF。我可以使用构造函数中的figsize参数调整第一个图形，但是如何更改第二个绘图的大小？

Answer 1

我使用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()

Answer 2

您可以使用gridspec和图：

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

Answer 3

• 另一种方法是使用subplot函数，并使用gridspec\u kw
  • matplotlib教程：使用GridSpec和其他函数自定义地物布局
  • matplotlib.gridspec.gridspec具有可用的gridspect\u 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')
  

  

  • 因为这个问题是规范的，所以这里有一个垂直子图的例子

  # plot it
  f, (a0, a1, a2) = plt.subplots(3, 1, gridspec_kw={'height_ratios': [1, 1, 3]})
  
  a0.plot(x, y)
  a1.plot(x, y)
  a2.plot(x, y)
  
  f.tight_layout()