将多个无花果从循环保存到一个pdf

Question

我有一个for循环用于绘制一组数字，我希望有一个pdf作为所有这些数字的结果。我正在使用plt.show()，但我不确定将fig.savefig('file_name.pdf')放在哪里或以不同的方式进行操作。现在它只是给了我最后一个数字。

我的代码：

  dpi = 100
  margin = 0.05
  xpixels, ypixels = 300, 600

for i in range(1,17):

  chr = pd.read_csv("chr"+str(i)+".txt", sep="\s+", header=None)

  print(chr.shape)

  figsize = (1 + margin) * ypixels / dpi, (1 + margin) * xpixels / dpi
  fig = plt.figure(figsize = figsize, dpi = dpi)

  ax = fig.add_axes([margin, margin, 1 - 2*margin, 1 - 2*margin])
  plt.ylabel("individuals")
  plt.title("Chromosome "+ str(i))

  ax.imshow(chr.values, interpolation='nearest', aspect = 4, cmap = plt.get_cmap('winter'))
  plt.show()

  fig.savefig('file_name.pdf')

感谢您的任何想法。 ：）