У меня есть этот код для создания нескольких графиков из всех текстовых файлов в папке. Он отлично работает и показывает графики, но я не могу понять, как сохранить их все.

import re
import numpy as np
import matplotlib.pyplot as plt
import pylab as pl
import os

rootdir='C:\documents\Neighbors for each search id'

for subdir,dirs,files in os.walk(rootdir):
 for file in files:
  f=open(os.path.join(subdir,file),'r')
  print file
  data=np.loadtxt(f)

  #plot data
  pl.plot(data[:,1], data[:,2], 'gs')

  #Put in the errors
  pl.errorbar(data[:,1], data[:,2], data[:,3], data[:,4], fmt='ro')

  #Dashed lines showing pmRa=0 and pmDec=0
  pl.axvline(0,linestyle='--', color='k')
  pl.axhline(0,linestyle='--', color='k')
  pl.show()

  f.close()

Я ранее использовал

fileName="C:\documents\FirstPlot.png"
plt.savefig(fileName, format="png")

но я думаю, что это просто сохраняет каждый график в один файл и перезаписывает последний.