Matplotlib PDF图无法响应figsize

Question

我希望以下代码确定子图的大小，以使生成的PDF宽度为5英寸，高度为8英寸。但是，无论我插入什么figsize，生成的文件都是8英寸宽和6英寸高。我究竟做错了什么？

import matplotlib.pyplot as plt
import matplotlib.gridspec as gs

fig = plt.Figure(figsize=(5,8))
fig.set_canvas(plt.gcf().canvas)

gs1 = gs.GridSpec(3,2)
gs1.update(wspace=0.4,hspace=0.4)
ax1 = plt.subplot(gs1[0,0])
ax2 = plt.subplot(gs1[0,1])
ax3 = plt.subplot(gs1[1,0])
ax4 = plt.subplot(gs1[1,1])
ax5 = plt.subplot(gs1[2,:])

ax1.plot([1,2,3],[4,5,6], 'k-')

fig.savefig("foo.pdf", format='pdf')

糟糕，编辑过后，我也曾尝试过fig.set_size_inches((5,8))，但这似乎也没有任何效果。

Answer 1

您可能会发现使用起来更方便matplotlib.pyplot.figure

在使用如下代码创建图形后尝试配置图形宽度

fig = plt.figure()
fig.set_figheight(5)
fig.set_figwidth(8)

我可能调换了尺寸，但这对我有用。这是从 matplotlib 文档中抄袭的完整示例，并对图形大小进行了修改。这也适用于figsizefigure()调用的参数。

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
fig.set_figheight(10)
fig.set_figwidth(12)
ax = fig.gca(projection='3d')
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
        linewidth=0, antialiased=False)
ax.set_zlim(-1.01, 1.01)

ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

fig.colorbar(surf, shrink=0.5, aspect=5)
fig.savefig("myfig.png", dpi=600) # useful for hi-res graphics
plt.show()