PdfPages在Matplotlib保存同一图两次

Question

下面的代码提供in the matplotlib documentation创建Hinton的图：

def hinton(matrix, max_weight=None, ax=None): 
    """Draw Hinton diagram for visualizing a weight matrix.""" 
    ax = ax if ax is not None else plt.gca() 

    if not max_weight: 
     max_weight = 2 ** np.ceil(np.log(np.abs(matrix).max())/np.log(2)) 

    ax.patch.set_facecolor('gray') 
    ax.set_aspect('equal', 'box') 
    ax.xaxis.set_major_locator(pl.NullLocator()) 
    ax.yaxis.set_major_locator(pl.NullLocator()) 

    for (x, y), w in np.ndenumerate(matrix): 
     color = 'white' if w > 0 else 'black' 
     size = np.sqrt(np.abs(w)/max_weight) 
     rect = pl.Rectangle([x - size/2, y - size/2], size, size, 
          facecolor=color, edgecolor=color) 
     ax.add_patch(rect) 

    ax.autoscale_view() 
    ax.invert_yaxis() 

我想创建两个Hinton的图表：一个用于权重从输入要隐藏层，和一个从我的单层MLP中的隐藏层到输出层。我曾尝试（based on this jupyter notebook）：

W = model_created.layers[0].kernel.get_value(borrow=True) 
W = np.squeeze(W) 
print("W shape : ", W.shape) # (153, 15) 

W_out = model_created.layers[1].kernel.get_value(borrow=True) 
W_out = np.squeeze(W_out) 
print('W_out shape : ', W_out.shape) # (15, 8) 

with PdfPages('hinton_again.pdf') as pdf: 
    h1 = hinton(W) 
    h2 = hinton(W_out) 
    pdf.savefig() 

我也曾尝试（based on this answer）：

h1 = hinton(W) 
h2 = hinton(W_out) 

pp = PdfPages('hinton_both.pdf') 
pp.savefig(h1) 
pp.savefig(h2) 
pp.close() 

无论如何，结局都是一样的：W的韩丁图绘制得到两次。我怎样才能在我的第一套权重中加入欣顿图，并在同一pdf中加入我的第二组权重的欣顿图（如果我能得到两张欣顿图，我还将解决两个单独的pdf）？

Answer 1

pdf.savefig()命令保存当前的数字。由于只有一个当前数字，它会将其保存两次。为了得到两个数字，他们需要被创建，例如，通过plt.figure(1)和plt.figure(2)。

with PdfPages('hinton_again.pdf') as pdf: 
    plt.figure(1) 
    h1 = hinton(W) 
    pdf.savefig() 
    plt.figure(2) 
    h2 = hinton(W_out) 
    pdf.savefig() 

当然也有吨不同的方式来挽救两个地块，anotherone可能是

fig, ax = plt.subplots() 
hinton(W, ax=ax) 

fig2, ax2 = plt.subplots() 
hinton(W_out, ax=ax2) 

with PdfPages('hinton_again.pdf') as pdf: 
    pdf.savefig(fig) 
    pdf.savefig(fig2)