def save(self,fig,p2p,fname): utils.trycreate(p2p) fpath = os.path.join(p2p,fname) #self.fig.savefig(fpath) plt.gcf().savefig(fpath,bbox_inches='tight')
if isinstance(y, collections.Sequence) and not isinstance(y, basestring): return x else: return [x] def plot_cross_section(self,var,latA,lonA,latB,lonB): xs = CrossSection() xs.plot(var,latA,lonA,latB,lonB) def save_data(self,data,folder,fname) # Ensure file suffix will be .npy os.path.splitext(fname)[0] # Check for folder and create if necessary utils.trycreate(folder) # Save in binary fpath = os.path.join(folder,fname) N.save(fpath,data) print("Save file {0}.npy to {1}.".format(fname,folder)) def compute_diff_energy( self,ptype,energy,files,times,upper=None,lower=None, d_save=1,d_return=1,d_fname='diff_energy_data'): """ This method computes difference kinetic energy (DKE) or different total energy (DTE, including temp) between WRFout files for a given depth of the atmosphere, at given time intervals