f_saving_plots.py 653 B

123456789101112131415161718192021
  1. import pylab as pl
  2. import numpy as np
  3. from matplotlib.backends.backend_pdf import PdfPages
  4. import matplotlib.pyplot as plt
  5. import pickle
  6. def saving_pdf_figure(fig,fig_name,md_dir,pickle_f=True):
  7. ax=plt.figure()
  8. ax=fig
  9. ax2=fig
  10. pdffig = PdfPages(fig_name+".pdf")
  11. ax.savefig(fig_name+'.png',dpi=200)
  12. fig.savefig(pdffig, format="pdf")
  13. metadata = pdffig.infodict()
  14. metadata['Title'] = md_dir['Title']
  15. metadata['Author'] = md_dir['Author']
  16. metadata['Subject'] = md_dir['Subject']
  17. metadata['Keywords'] = md_dir['Keywords']
  18. pdffig.close()
  19. if pickle_f:
  20. pickle.dump(ax2, file(fig_name+'.pickle', 'w'))