Matplotlib - 网格排列子图间距

Question

我正在尝试像时尚一样将一堆子图排列在网格中。问题在于，子图的数量随着用户选择绘制的数据而变化。现在我正在尝试以这种方式添加绘图：

    l = len(clicked)

    self.p=wx.Panel(self)
    self.dpi=100
    self.fig = Figure()
    self.canvas = FigCanvas(self.p, -1, self.fig)
    if l == 1:
        self.splt = self.fig.add_subplot(1,1,1)
        self.plt=self.splt.plot(np.arange(5),np.arange(5))
    else:
        for i in np.arange(l):
            self.splt=self.fig.add_subplot(l+1,2,i+1)
            self.fig.subplots_adjust(left=0.7, bottom=0.6, right=0.75, top=0.75)
            self.plt=self.splt.plot(np.arange(5),np.arange(5))

我只是使用假数据进行调试。无论如何，我正在使用 wxPython 在框架内绘制它。这里的clicked提供了用户所做选择的数量。我已经尝试过使用 subplots_adjust() ，结果与我想要的完全相反。情节被缩小得难以辨认。有没有办法将绘图排列在某种网格中。我看到有一个选项 subplot2grid，但我还没有让它处理可变数量的子图。

Answer 1

希望这会有所帮助：

from __future__ import division
import numpy as np
import pylab as plt

def divideSquare(N):
    if N<1:
        return 1,1

    minX = max(1,int(np.floor(np.sqrt(N))))
    maxX = max(minX+1, int(np.ceil(np.sqrt(5.0/3*N))))
    Nx = range(minX, maxX+1)
    Ny = [np.ceil(N/y) for y in Nx]
    err = [np.uint8(y*x - N) for y in Nx for x in Ny]
    ind = np.argmin(err)
    y = Nx[int(ind/len(Ny))]
    x = Ny[ind%len(Nx)]
    return min(y,x), max(y,x)

Nlist = np.arange(0,101)
empty = np.zeros_like(Nlist)
ratio = np.zeros_like(Nlist, dtype = float)
for k, N in enumerate(Nlist):
    y,x = divideSquare(N)
    empty[k] = y*x - N
    ratio[k] = 1.0 * x/y
    print y,x,y/x

plt.figure(1)
plt.clf()
y,x = divideSquare(2)
plt.subplot(y,x,1)    
plt.plot(Nlist, empty, 'k')
plt.xlabel('Number of plots')
plt.ylabel('Empty squares left')
plt.subplot(y,x,2)
plt.plot(Nlist, ratio, 'r')
plt.xlabel('Number of plots')
plt.ylabel('Ratio')

def divide2or3(N):
    if N<1:
        return 1,1
    if N%2==0:
        return int(np.ceil(N/2)), 2
    return int(np.ceil(N/3)), 3

Nlist = np.arange(0,101)
empty = np.zeros_like(Nlist)
ratio = np.zeros_like(Nlist, dtype = float)
for k, N in enumerate(Nlist):
    y,x = divide2or3(N)
    empty[k] = y*x - N
    ratio[k] = 1.0 * x/y
    print y,x,y/x

plt.figure(2)
plt.clf()
y,x = divide2or3(2)
plt.subplot(y,x,1)    
plt.plot(Nlist, empty, 'k')
plt.xlabel('Number of plots')
plt.ylabel('Empty squares left')
plt.subplot(y,x,2)
plt.plot(Nlist, ratio, 'r')
plt.xlabel('Number of plots')
plt.ylabel('Ratio')

plt.show()

divideSquare(N) 尝试获得接近 1 的比率，但空仓数量最少。
divide2or3(N) 为您提供 2 或 3 列的网格（如果反转 x 和 y，则为行），但我不确定这是否是您想要的 90 个图。