start_time = time.time() othercon = 'Profile_Time >= "{}" and Profile_Time <"{}" and high_level > 338'.format(desday,tom_dt.strftime('%Y-%m-%d')) # apro_df 是[latitude,longitude,time,high_level，features]的格式，但是高度还没有std apro_ori, apro_df, apro_xr = get_apro_data_sql(con, apro_config, othercon, pos_merge=pos_df, multi_index=multi_index + ['high_level']) print('THE COST to get raw data table:',time.strftime("%H: %M: %S",time.gmtime(time.time() - start_time))) # TODO: 可能查不到数据，判断一下 if apro_df.shape[0] == 0: # 修改列名即可 apro_final_df = apro_df apro_final_df.rename(columns={'high_level':'Level'},inplace=True) print('THE {} DAY HAS NO APRO DATA'.format(desday)) else: # 高度标准化 apro_df['Level'] = apro_df.apply(apro_get_level, axis=1) apro_df = apro_df.drop(['high_level'], axis=1) apro_xr = apro_df.set_index(['Time', 'Latitude', 'Longitude', 'Level']).to_xarray() # 插值等 # 2. 插值 _, _, times, tlabels = get_apro_interp_attr(apro_xr, std_index_3d, desday,posrange) # 时间 apro_mean_xr = apro_xr.groupby_bins('Time', bins=times, labels=tlabels).mean('Time').rename( {'Time_bins': 'Time'}) # 位置 apro_mean_xr['Latitude'] = apro_mean_xr.Latitude.values.round(1) apro_mean_xr['Longitude'] = apro_mean_xr.Longitude.values.round(1) apro_mean_df = apro_mean_xr.to_dataframe().dropna(how='all').reset_index() # 最后 apro_final_df = apro_mean_df.groupby(['Time', 'Latitude', 'Longitude', 'Level']).mean().dropna(how='all') # apro_final_xr = apro_final_df.to_xarray() apro_final_df = apro_final_df.reset_index() # 修改时间 apro_final_df.Time = pd.to_datetime(apro_final_df['Time']) apro_final_df.Time = apro_final_df['Time'].apply(lambda x:x.replace(year=2023)) # Todo: 可以改成输入的年份 # 输出中间文件,可能是空文件 desday = desday.replace('2017','2023') outfile = os.path.join(apro_config.outpath,"apro_mid_{}.csv".format(desday)) apro_final_df.to_csv(outfile,index=False)