reader = ReadClass() remote_data = reader.data_PID(setting_name) reader.set_ssh(remote_data=remote_data) reader.read_setting(setting_name) reader.read_file_number(ssh_flag=True) budget = reader.read_budget(number_time=420, ssh_flag=False) columns = [ 'T', 'E', 'E-first', 'E-diag', 'E-lower', 'E-upper', 'Q2', 'Q2-first', 'Q2-diag', 'Q2-lower', 'Q2-upper', 'PE' ] df = pd.DataFrame(data=budget, columns=columns) df = df.set_index('T') for col in df.columns.tolist(): df = df.rename(columns={col: reader.long_name(col)}) df['Q2-others'] = df[reader.long_name('Q2')] - df[reader.long_name('Q2-first')] df['Q2-others-mean'] = df['Q2-others'] / (reader.N - 1) df.plot( y=[reader.long_name('Q2-first'), reader.long_name('E-first'), 'Q2-others'], # reader.long_name('Q2-lower'), # reader.long_name('Q2-upper')], logy=True, # xlim=xlim, # ylim=ylim figsize=(9, 3)) plt.savefig(figure_dir + 'Enstrophy_total' + setting_name + '.eps') plt.show()
reader = ReadClass() remote_data = reader.data_PID(setting_name) reader.set_ssh(remote_data=remote_data) reader.read_setting(setting_name) reader.read_file_number(ssh_flag=True) budget = reader.read_budget(number_time=420, ssh_flag=True) columns = [ 'T', 'E', 'E-first', 'E-diag', 'E-lower', 'E-upper', 'Q2', 'Q2-first', 'Q2-diag', 'Q2-lower', 'Q2-upper', 'PE' ] df = pd.DataFrame(data=budget, columns=columns) df = df.set_index('T') for col in df.columns.tolist(): df = df.rename(columns={col: reader.long_name(col)}) df['E-others'] = df[reader.long_name('E')] - df[reader.long_name('E-first')] df.plot( y=[ reader.long_name('E'), reader.long_name('E-first'), 'E-others', reader.long_name('E-lower'), reader.long_name('E-upper') ], # logy=True, # xlim=xlim, ylim=[0, df[reader.long_name('E')].max() * 1.1], figsize=(9, 3)) plt.savefig(figure_dir + 'Energy_' + setting_name + '.eps')
reader = ReadClass() remote_data = reader.data_PID(setting_name) reader.set_ssh(remote_data=remote_data) reader.read_setting(setting_name) reader.read_file_number(ssh_flag=True) budget = reader.read_budget() columns = [ 'T', 'E', 'E-diag', 'E-lower', 'E-upper', 'Q2', 'Q2-diag', 'Q2-lower', 'Q2-upper', 'PE' ] df = pd.DataFrame(data=budget, columns=columns) df = df.set_index('T') for col in df.columns.tolist(): df = df.rename(columns={col: reader.long_name(col)}) df.plot( y=[ reader.long_name('E'), reader.long_name('E-diag'), reader.long_name('E-lower'), reader.long_name('E-upper') ], # logy=True, figsize=(9, 3) # xlim=xlim, # ylim=ylim ) plt.savefig(figure_dir + 'Energy' + setting_name + '.eps')
ylim = [0.0028, 0.003] reader = ReadClass() remote_data = reader.data_PID(setting_name) reader.set_ssh(remote_data=remote_data) reader.read_setting(setting_name) reader.read_file_number(ssh_flag=True) budget = reader.read_budget(number_time=118, ssh_flag=True) columns = ['T', 'E', 'E-diag', 'E-lower', 'E-upper', 'Q2', 'Q2-diag', 'Q2-lower', 'Q2-upper', 'PE'] df = pd.DataFrame(data=budget, columns=columns) df = df.set_index('T') for col in df.columns.tolist(): df = df.rename(columns={col: reader.long_name(col)}) df.plot(y=[reader.long_name('E')], # reader.long_name('E-diag'), # reader.long_name('E-lower'), # reader.long_name('E-upper')], # logy=True, # xlim=xlim, # ylim=ylim, figsize=(9, 3) ) plt.savefig(figure_dir + 'Energy' + setting_name + '.eps') plt.show() plt.close()