clean

scripts/01_crabmodel_plot_poly.py

@@ -6,7 +6,7 @@ __copyright__   = "Space Research Institute (IKI)"
 from astropy.table import Table, Column
 
 import matplotlib.pyplot as plt
-import sys
+import sys, os
 
 from ridge.utils import *
 from ridge.config import *
@@ -14,15 +14,15 @@ from ridge.config import *
 
 scale = 1e-3
 
-fn="detcnts.E01.crabmodel.fits"
+fn="detcnts.A01.crabmodel.fits"
 dat = Table.read(proddir+fn, unit_parse_strict='silent')
 df1 = dat.to_pandas().sort_values(by=['REV'])
 
-fn="detcnts.E14.crabmodel.fits"
+fn="detcnts.A02.crabmodel.fits"
 dat = Table.read(proddir+fn, unit_parse_strict='silent')
 df2 = dat.to_pandas().sort_values(by=['REV'])
 
-fn="detcnts.E13.crabmodel.fits"
+fn="detcnts.A03.crabmodel.fits"
 dat = Table.read(proddir+fn, unit_parse_strict='silent')
 df3 = dat.to_pandas().sort_values(by=['REV'])
 
@@ -69,5 +69,10 @@ ax2.set_ylabel('Count rate, x10$^{-3}$ cts s$^{-1}$ pix$^{-1}$',fontsize=14, fon
 #plt.xscale('linear')
 #plt.yscale('linear')
 
-plt.savefig(proddir+'crabmodel_poly.png', bbox_inches='tight')
+if not os.path.exists(figdir):
+    os.makedirs(figdir)
+
+filename=figdir+'crabmodel_poly.png'
+plt.savefig(filename, bbox_inches='tight')
 plt.close(fig)
+print("Result is saved as {}".format(filename))