python - 子图中的分页符？多个页面上的 Matplotlib 子图

Question

我想创建一个能够将多个图形绘制到一个 PDF 文件中的 python 程序，但是子图的数量是可变的。我已经做到了，每页一个图。然而，因为我得到了 someteimes around 100 plots 这使得很多滚动并且没有真正清楚地显示。因此，我希望每页有 5X4 个子图。我已经为此编写了代码，整个代码很长，而且由于我是 pyhton 的新手，对于知道该怎么做的人来说它看起来很糟糕，但是绘图部分看起来像这样:

rows = (len(tags))/5
fig = plt.figure()
count = 0    
for keyInTags in tags:
       count = count + 1
       ax = fig.add_subplot(int(rows), 5, count)
       ax.set_title("cell" + keyInTags)
       ax.plot(x, y_green, color='k')
       ax.plot(x, y_red, color='k')
       plt.subplots_adjust(hspace=0.5, wspace=0.3)
pdf.savefig(fig)

我的想法是我得到一个 PDF，其中绘制了所有“细胞”(用于生物学研究)。到目前为止，我编写的代码运行良好，但是如果我得到超过 4 行的子图，我想做一个“pageprake”。在某些情况下，我在一页上有超过 21 行，这使得我无法看到任何东西。

那么，是否有解决方案，例如，告诉 Python 在 4 行后进行分页？在 id 为 21 行的情况下，有 6 个页面有漂亮的可见图。还是通过绘制 5x4 图然后以某种方式遍历文件来完成？如果有人能提供一点帮助或提示，我会很高兴。我在这里坐了 4 个小时，没有找到解决方案。

Answer 1

A.遍历页面

您可以找出需要多少页 (npages) 并为每页创建一个新图形。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages


tags = ["".join(np.random.choice(list("ABCDEFG123"), size=5)) for _ in range(53)]

N = len(tags)  # number of subplots
nrows = 5      # number of rows per page
ncols = 4      # number of columns per page

# calculate number of pages needed
npages = N // (nrows*ncols)
if N % (nrows*ncols) > 0:
    npages += 1

pdf = PdfPages('out2.pdf')

for page in range(npages):
    fig = plt.figure(figsize=(8,11))
    for i in range(min(nrows*ncols, N-page*(nrows*ncols))):
        # Your plot here
        count = page*ncols*nrows+i
        ax = fig.add_subplot(nrows, ncols, i+1)
        ax.set_title(f"{count} - {tags[count]}")
        ax.plot(np.cumsum(np.random.randn(33)))
        # end of plotting

    fig.tight_layout()
    pdf.savefig(fig)

pdf.close()
plt.show()

B.循环数据

或者您也可以遍历标签本身并在需要时创建一个新图形:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages


tags = ["".join(np.random.choice(list("ABCDEFG123"), size=5)) for _ in range(53)]

nrows = 5      # number of rows per page
ncols = 4      # number of columns per page

pdf = PdfPages('out2.pdf')

for i, tag in enumerate(tags):
    j = i % (nrows*ncols)
    if j == 0:
        fig = plt.figure(figsize=(8,11))

    ax = fig.add_subplot(nrows, ncols,j+1)
    ax.set_title(f"{i} - {tags[i]}")
    ax.plot(np.cumsum(np.random.randn(33)))
    # end of plotting
    if j == (nrows*ncols)-1 or i == len(tags)-1:
       fig.tight_layout()
       pdf.savefig(fig)

pdf.close()
plt.show()