def main(sDir, url_list, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
if preferred_only == 'yes':
ps_df, n_streams = cf.get_preferred_stream_info(r)
fdatasets = []
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, fdatasets)
num_data = len(fdatasets)
save_dir = os.path.join(sDir, array, subsite, r,
'preferred_method_plots')
cf.create_dir(save_dir)
print(len(fdatasets))
if len(fdatasets) > 3:
steps = list(range(3, len(fdatasets) + 3, 3))
for ii in steps:
plot_velocity_variables(r, fdatasets[ii - 3:ii], 3, save_dir)
else:
plot_velocity_variables(r, fdatasets, 3, save_dir)
def main(sDir, url_list, start_time, end_time, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
fdatasets = np.unique(fdatasets).tolist()
for fd in fdatasets:
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
#ds_vars = list(ds.data_vars.keys()) + [x for x in ds.coords.keys() if 'pressure' in x] # get pressure variable from coordinates
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'.
format(start_time, end_time))
continue
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(
fd)
sci_vars = cf.return_science_vars(stream)
# drop the following list of key words from science variables list
sci_vars = notin_list(
sci_vars, ['bin_depths', 'salinity', 'temperature', 'beam'])
sci_vars = [
name for name in sci_vars if ds[name].units != 'mm s-1'
]
print('\nPlotting {} {}'.format(r, deployment))
array = subsite[0:2]
filename = '_'.join(fname.split('_')[:-1])
save_dir = os.path.join(sDir, array, subsite, refdes,
'preferred_method_plots', deployment)
cf.create_dir(save_dir)
tm = ds['time'].values
t0 = pd.to_datetime(tm.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(tm.max()).strftime('%Y-%m-%dT%H:%M:%S')
title_text = ' '.join((deployment, refdes, method))
for var in sci_vars:
print(var)
v = ds[var]
fv = v._FillValue
v_name = v.long_name
if len(v.dims) == 1:
v, n_nan, n_fv, n_ev, n_grange, g_min, g_max, n_std = reject_err_data_1_dims(
v, fv, r, var, n=5)
# Plot all data
fig, ax = pf.plot_timeseries(tm, v, v_name, stdev=None)
ax.set_title((title_text + '\n' + t0 + ' - ' + t1),
fontsize=9)
sfile = '-'.join((filename, v_name, t0[:10]))
pf.save_fig(save_dir, sfile)
# Plot data with outliers removed
fig, ax = pf.plot_timeseries(tm, v, v_name, stdev=5)
title_i = 'removed: {} nans, {} fill values, {} extreme values, {} GR [{}, {}],' \
' {} outliers +/- 5 SD'.format(n_nan, n_fv , n_ev, n_grange, g_min, g_max, n_std)
ax.set_title(
(title_text + '\n' + t0 + ' - ' + t1 + '\n' + title_i),
fontsize=8)
sfile = '-'.join(
(filename, v_name, t0[:10])) + '_rmoutliers'
pf.save_fig(save_dir, sfile)
else:
v = v.values.T.astype(float)
v_bad_beams = ds[
'percent_bad_beams'] # get bad beams percent
fv_bad_beam = v_bad_beams._FillValue
v_bad_beams = v_bad_beams.values.T.astype(float)
v_bad_beams[v_bad_beams ==
fv_bad_beam] = np.nan # mask fill values
v, n_nan, n_fv, n_ev, n_bb, n_grange, g_min, g_max = reject_err_data_2_dims(
v, v_bad_beams, fv, r, var)
ylabel = 'bin_depths ({})'.format(ds['bin_depths'].units)
clabel = '{} ({})'.format(var, ds[var].units)
# check bin depths for extreme values
y = ds['bin_depths'].values.T
y_nan = np.sum(np.isnan(y))
y = np.where(y < 6000, y,
np.nan) # replace extreme bin_depths by nans
bin_nan = np.sum(np.isnan(y)) - y_nan
bin_title = 'removed: {} bin depths > 6000'.format(bin_nan)
if 'echo' in var:
color = 'BuGn'
else:
color = 'RdBu'
new_y = dropna(y,
axis=1) # convert to DataFrame to drop nan
y_mask = new_y.loc[list(new_y.index), list(new_y.columns)]
v_new = pd.DataFrame(v)
v_mask = v_new.loc[list(new_y.index), list(new_y.columns)]
tm_mask = tm[new_y.columns]
fig, ax, __ = pf.plot_adcp(tm_mask,
np.array(y_mask),
np.array(v_mask),
ylabel,
clabel,
color,
n_stdev=None)
if bin_nan > 0:
ax.set_title((title_text + '\n' + t0 + ' - ' + t1 +
'\n' + bin_title),
fontsize=8)
else:
ax.set_title((title_text + '\n' + t0 + ' - ' + t1),
fontsize=8)
sfile = '-'.join((filename, var, t0[:10]))
pf.save_fig(save_dir, sfile)
fig, ax, n_nans_all = pf.plot_adcp(tm_mask,
np.array(y_mask),
np.array(v_mask),
ylabel,
clabel,
color,
n_stdev=5)
title_i = 'removed: {} nans {} fill values, {} extreme values, {} bad beams, {} GR [{}, {}]'.format(
n_nan, n_fv, n_ev, n_bb, n_grange, g_min, g_max)
if bin_nan > 0:
ax.set_title((title_text + '\n' + t0 + ' - ' + t1 +
'\n' + title_i + '\n' + bin_title),
fontsize=8)
else:
ax.set_title((title_text + '\n' + t0 + ' - ' + t1 +
'\n' + title_i),
fontsize=8)
sfile = '-'.join((filename, var, t0[:10])) + '_rmoutliers'
pf.save_fig(save_dir, sfile)
def main(url_list, sDir, deployment_num, start_time, end_time, preferred_only, n_std, inpercentile, zcell_size, zdbar):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list and 'ENG' not in rd and 'ADCP' not in rd:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(main_sensor, fdatasets)
for fd in fdatasets_sel:
part_d = fd.split('/')[-1]
print('\n{}'.format(part_d))
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(fd)
array = subsite[0:2]
sci_vars = cf.return_science_vars(stream)
# if 'CE05MOAS' in r or 'CP05MOAS' in r: # for coastal gliders, get m_water_depth for bathymetry
# eng = '-'.join((r.split('-')[0], r.split('-')[1], '00-ENG000000', method, 'glider_eng'))
# eng_url = [s for s in url_list if eng in s]
# if len(eng_url) == 1:
# eng_datasets = cf.get_nc_urls(eng_url)
# # filter out collocated datasets
# eng_dataset = [j for j in eng_datasets if (eng in j.split('/')[-1] and deployment in j.split('/')[-1])]
# if len(eng_dataset) > 0:
# ds_eng = xr.open_dataset(eng_dataset[0], mask_and_scale=False)
# t_eng = ds_eng['time'].values
# m_water_depth = ds_eng['m_water_depth'].values
#
# # m_altitude = glider height above seafloor
# # m_depth = glider depth in the water column
# # m_altitude = ds_eng['m_altitude'].values
# # m_depth = ds_eng['m_depth'].values
# # calc_water_depth = m_altitude + m_depth
#
# # m_altimeter_status = 0 means a good reading (not nan or -1)
# try:
# eng_ind = ds_eng['m_altimeter_status'].values == 0
# except KeyError:
# eng_ind = (~np.isnan(m_water_depth)) & (m_water_depth >= 0)
#
# m_water_depth = m_water_depth[eng_ind]
# t_eng = t_eng[eng_ind]
#
# # get rid of any remaining nans or fill values
# eng_ind2 = (~np.isnan(m_water_depth)) & (m_water_depth >= 0)
# m_water_depth = m_water_depth[eng_ind2]
# t_eng = t_eng[eng_ind2]
# else:
# print('No engineering file for deployment {}'.format(deployment))
# m_water_depth = None
# t_eng = None
# else:
# m_water_depth = None
# t_eng = None
# else:
# m_water_depth = None
# t_eng = None
if deployment_num is not None:
if int(int(deployment[-4:])) is not deployment_num:
print(type(int(deployment[-4:])), type(deployment_num))
continue
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print('No data to plot for specified time range: ({} to {})'.format(start_time, end_time))
continue
stime = start_time.strftime('%Y-%m-%d')
etime = end_time.strftime('%Y-%m-%d')
ext = stime + 'to' + etime # .join((ds0_method, ds1_method
save_dir_profile = os.path.join(sDir, array, subsite, refdes, 'profile_plots', deployment, ext)
save_dir_xsection = os.path.join(sDir, array, subsite, refdes, 'xsection_plots', deployment, ext)
save_dir_4d = os.path.join(sDir, array, subsite, refdes, 'xsection_plots_4d', deployment, ext)
else:
save_dir_profile = os.path.join(sDir, array, subsite, refdes, 'profile_plots', deployment)
save_dir_xsection = os.path.join(sDir, array, subsite, refdes, 'xsection_plots', deployment)
save_dir_4d = os.path.join(sDir, array, subsite, refdes, 'xsection_plots_4d', deployment)
time1 = ds['time'].values
try:
ds_lat1 = ds['lat'].values
except KeyError:
ds_lat1 = None
print('No latitude variable in file')
try:
ds_lon1 = ds['lon'].values
except KeyError:
ds_lon1 = None
print('No longitude variable in file')
# get pressure variable
pvarname, y1, y_units, press, y_fillvalue = cf.add_pressure_to_dictionary_of_sci_vars(ds)
for sv in sci_vars:
print('')
print(sv)
if 'pressure' not in sv:
if sv == 'spkir_abj_cspp_downwelling_vector':
pxso.pf_xs_spkir(ds, sv, time1, y1, ds_lat1, ds_lon1, zcell_size, inpercentile, save_dir_profile,
save_dir_xsection, deployment, press, y_units, n_std, zdbar)
elif 'OPTAA' in r:
if sv not in ['wavelength_a', 'wavelength_c']:
pxso.pf_xs_optaa(ds, sv, time1, y1, ds_lat1, ds_lon1, zcell_size, inpercentile, save_dir_profile,
save_dir_xsection, deployment, press, y_units, n_std, zdbar)
else:
z1 = ds[sv].values
fv = ds[sv]._FillValue
sv_units = ds[sv].units
# Check if the array is all NaNs
if sum(np.isnan(z1)) == len(z1):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z1[z1 != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
# remove unreasonable pressure data (e.g. for surface piercing profilers)
if zdbar:
po_ind = (0 < y1) & (y1 < zdbar)
tm = time1[po_ind]
y = y1[po_ind]
z = z1[po_ind]
ds_lat = ds_lat1[po_ind]
ds_lon = ds_lon1[po_ind]
else:
tm = time1
y = y1
z = z1
ds_lat = ds_lat1
ds_lon = ds_lon1
# reject erroneous data
dtime, zpressure, ndata, lenfv, lennan, lenev, lengr, global_min, global_max, lat, lon = \
cf.reject_erroneous_data(r, sv, tm, y, z, fv, ds_lat, ds_lon)
# get rid of 0.0 data
if sv == 'salinity':
ind = ndata > 30
elif sv == 'density':
ind = ndata > 1022.5
elif sv == 'conductivity':
ind = ndata > 3.45
else:
ind = ndata > 0
# if sv == 'sci_flbbcd_chlor_units':
# ind = ndata < 7.5
# elif sv == 'sci_flbbcd_cdom_units':
# ind = ndata < 25
# else:
# ind = ndata > 0.0
# if 'CTD' in r:
# ind = zpressure > 0.0
# else:
# ind = ndata > 0.0
lenzero = np.sum(~ind)
dtime = dtime[ind]
zpressure = zpressure[ind]
ndata = ndata[ind]
if ds_lat is not None and ds_lon is not None:
lat = lat[ind]
lon = lon[ind]
else:
lat = None
lon = None
if len(dtime) > 0:
# reject time range from data portal file export
t_portal, z_portal, y_portal, lat_portal, lon_portal = \
cf.reject_timestamps_dataportal(subsite, r, dtime, zpressure, ndata, lat, lon)
print('removed {} data points using visual inspection of data'.format(
len(ndata) - len(z_portal)))
# create data groups
if len(y_portal) > 0:
columns = ['tsec', 'dbar', str(sv)]
min_r = int(round(np.nanmin(y_portal) - zcell_size))
max_r = int(round(np.nanmax(y_portal) + zcell_size))
ranges = list(range(min_r, max_r, zcell_size))
groups, d_groups = gt.group_by_depth_range(t_portal, y_portal, z_portal, columns, ranges)
if 'scatter' in sv:
n_std = None # to use percentile
else:
n_std = n_std
# get percentile analysis for printing on the profile plot
y_avg, n_avg, n_min, n_max, n0_std, n1_std, l_arr, time_ex = cf.reject_timestamps_in_groups(
groups, d_groups, n_std, inpercentile)
"""
Plot all data
"""
if len(time1) > 0:
cf.create_dir(save_dir_profile)
cf.create_dir(save_dir_xsection)
sname = '-'.join((r, method, sv))
sfileall = '_'.join(('all_data', sname, pd.to_datetime(time1.min()).strftime('%Y%m%d')))
tm0 = pd.to_datetime(time1.min()).strftime('%Y-%m-%dT%H:%M:%S')
tm1 = pd.to_datetime(time1.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method)) + '\n' + tm0 + ' to ' + tm1
if 'SPKIR' in r:
title = title + '\nWavelength = 510 nm'
'''
profile plot
'''
xlabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
clabel = 'Time'
fig, ax = pf.plot_profiles(z1, y1, time1, ylabel, xlabel, clabel, stdev=None)
ax.set_title(title, fontsize=9)
fig.tight_layout()
pf.save_fig(save_dir_profile, sfileall)
'''
xsection plot
'''
clabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
fig, ax, bar = pf.plot_xsection(subsite, time1, y1, z1, clabel, ylabel, t_eng=None,
m_water_depth=None, inpercentile=None, stdev=None)
if fig:
ax.set_title(title, fontsize=9)
fig.tight_layout()
pf.save_fig(save_dir_xsection, sfileall)
"""
Plot cleaned-up data
"""
if len(dtime) > 0:
if len(y_portal) > 0:
sfile = '_'.join(('rm_erroneous_data', sname, pd.to_datetime(t_portal.min()).strftime('%Y%m%d')))
t0 = pd.to_datetime(t_portal.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(t_portal.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method)) + '\n' + t0 + ' to ' + t1
if 'SPKIR' in r:
title = title + '\nWavelength = 510 nm'
'''
profile plot
'''
xlabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
clabel = 'Time'
fig, ax = pf.plot_profiles(z_portal, y_portal, t_portal, ylabel, xlabel, clabel, stdev=None)
ax.set_title(title, fontsize=9)
ax.plot(n_avg, y_avg, '-k')
ax.fill_betweenx(y_avg, n0_std, n1_std, color='m', alpha=0.2)
if inpercentile:
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} unreasonable values'.format(lenfv, lennan, lenev, lengr, global_min, global_max, lenzero) +
'\nexcluded {} suspect data points when inspected visually'.format(
len(ndata) - len(z_portal)) +
'\n(black) data average in {} dbar segments'.format(zcell_size) +
'\n(magenta) {} percentile envelope in {} dbar segments'.format(
int(100 - inpercentile * 2), zcell_size),)
elif n_std:
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} unreasonable values'.format(lenfv, lennan, lenev, lengr, global_min, global_max,
lenzero) +
'\nexcluded {} suspect data points when inspected visually'.format(
len(ndata) - len(z_portal)) +
'\n(black) data average in {} dbar segments'.format(zcell_size) +
'\n(magenta) +/- {} SD envelope in {} dbar segments'.format(
int(n_std), zcell_size),)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
fig.tight_layout()
pf.save_fig(save_dir_profile, sfile)
'''
xsection plot
'''
clabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
# plot non-erroneous data
fig, ax, bar = pf.plot_xsection(subsite, t_portal, y_portal, z_portal, clabel, ylabel,
t_eng=None, m_water_depth=None, inpercentile=None,
stdev=None)
ax.set_title(title, fontsize=9)
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} unreasonable values'.format(lenfv, lennan, lenev, lengr, global_min, global_max, lenzero) +
'\nexcluded {} suspect data points when inspected visually'.format(
len(ndata) - len(z_portal)),
)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
fig.tight_layout()
pf.save_fig(save_dir_xsection, sfile)
'''
4D plot for gliders only
'''
if 'MOAS' in r:
if ds_lat is not None and ds_lon is not None:
cf.create_dir(save_dir_4d)
clabel = sv + " (" + sv_units + ")"
zlabel = press[0] + " (" + y_units[0] + ")"
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
sct = ax.scatter(lon_portal, lat_portal, y_portal, c=z_portal, s=2)
cbar = plt.colorbar(sct, label=clabel, extend='both')
cbar.ax.tick_params(labelsize=8)
ax.invert_zaxis()
ax.view_init(25, 32)
ax.invert_xaxis()
ax.invert_yaxis()
ax.set_zlabel(zlabel, fontsize=9)
ax.set_ylabel('Latitude', fontsize=9)
ax.set_xlabel('Longitude', fontsize=9)
ax.set_title(title, fontsize=9)
pf.save_fig(save_dir_4d, sfile)
def main(sDir, url_list, start_time, end_time, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
rms = '-'.join((r, row[ii]))
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(
main_sensor, fdatasets)
for fd in fdatasets_sel:
with xr.open_dataset(fd, mask_and_scale=False) as ds:
ds = ds.swap_dims({'obs': 'time'})
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'
.format(start_time, end_time))
continue
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(
fd)
print('\nPlotting {} {}'.format(r, deployment))
array = subsite[0:2]
save_dir = os.path.join(sDir, array, subsite, refdes,
'timeseries_panel_plots')
filename = '_'.join(fname.split('_')[:-1])
sci_vars = cf.return_science_vars(stream)
if len(sci_vars) > 1:
cf.create_dir(save_dir)
colors = cm.jet(np.linspace(0, 1, len(sci_vars)))
t = ds['time'].values
t0 = pd.to_datetime(t.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(t.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method))
# Plot data with outliers removed
fig, ax = pf.plot_timeseries_panel(ds, t, sci_vars, colors,
5)
plt.xticks(fontsize=7)
ax[0].set_title((title + '\n' + t0 + ' - ' + t1),
fontsize=7)
sfile = '-'.join((filename, 'timeseries_panel', t0[:10]))
pf.save_fig(save_dir, sfile)
else:
print(
'Only one science variable in file, no panel plots necessary'
)
def main(sDir, plotting_sDir, url_list, sd_calc):
dr = pd.read_csv('https://datareview.marine.rutgers.edu/notes/export')
drn = dr.loc[dr.type == 'exclusion']
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
pms = []
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
pms.append(row[ii])
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(
main_sensor, fdatasets)
# find time ranges to exclude from analysis for data review database
subsite = r.split('-')[0]
subsite_node = '-'.join((subsite, r.split('-')[1]))
drne = drn.loc[drn.reference_designator.isin(
[subsite, subsite_node, r])]
et = []
for i, row in drne.iterrows():
sdate = cf.format_dates(row.start_date)
edate = cf.format_dates(row.end_date)
et.append([sdate, edate])
# get science variable long names from the Data Review Database
stream_sci_vars = cd.sci_var_long_names(r)
# check if the science variable long names are the same for each stream
sci_vars_dict = cd.sci_var_long_names_check(stream_sci_vars)
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
# build dictionary of science data from the preferred dataset for each deployment
print('\nAppending data from files')
sci_vars_dict, pressure_unit, pressure_name = cd.append_science_data(
ps_df, n_streams, r, fdatasets_sel, sci_vars_dict, et)
# analyze combined dataset
print('\nAnalyzing combined dataset and writing summary file')
array = subsite[0:2]
save_dir = os.path.join(sDir, array, subsite)
cf.create_dir(save_dir)
rows = []
if ('FLM' in r) and (
'CTDMO' in r
): # calculate Flanking Mooring CTDMO stats based on pressure
headers = [
'common_stream_name', 'preferred_methods_streams',
'deployments', 'long_name', 'units', 't0', 't1', 'fill_value',
'global_ranges', 'n_all', 'press_min_max',
'n_excluded_forpress', 'n_nans', 'n_fillvalues', 'n_grange',
'define_stdev', 'n_outliers', 'n_stats', 'mean', 'min', 'max',
'stdev', 'note'
]
else:
headers = [
'common_stream_name', 'preferred_methods_streams',
'deployments', 'long_name', 'units', 't0', 't1', 'fill_value',
'global_ranges', 'n_all', 'n_nans', 'n_fillvalues', 'n_grange',
'define_stdev', 'n_outliers', 'n_stats', 'mean', 'min', 'max',
'stdev'
]
for m, n in sci_vars_dict.items():
print('\nSTREAM: ', m)
if m == 'common_stream_placeholder':
m = 'science_data_stream'
if m == 'metbk_hourly': # don't calculate ranges for metbk_hourly
continue
if ('FLM' in r) and (
'CTDMO' in r
): # calculate Flanking Mooring CTDMO stats based on pressure
# index the pressure variable to filter and calculate stats on the rest of the variables
sv_press = 'Seawater Pressure'
vinfo_press = n['vars'][sv_press]
# first, index where data are nans, fill values, and outside of global ranges
fv_press = list(np.unique(vinfo_press['fv']))[0]
pdata = vinfo_press['values']
[pind, __, __, __, __,
__] = index_dataset(r, vinfo_press['var_name'], pdata,
fv_press)
pdata_filtered = pdata[pind]
[__, pmean, __, __, psd,
__] = cf.variable_statistics(pdata_filtered, None)
# index of pressure = average of all 'valid' pressure data +/- 1 SD
ipress_min = pmean - psd
ipress_max = pmean + psd
ind_press = (pdata >= ipress_min) & (pdata <= ipress_max)
# calculate stats for all variables
print('\nPARAMETERS:')
for sv, vinfo in n['vars'].items():
print(sv)
fv_lst = np.unique(vinfo['fv']).tolist()
if len(fv_lst) == 1:
fill_value = fv_lst[0]
else:
print('No unique fill value for {}'.format(sv))
lunits = np.unique(vinfo['units']).tolist()
n_all = len(vinfo['t'])
# filter data based on pressure index
t_filtered = vinfo['t'][ind_press]
data_filtered = vinfo['values'][ind_press]
deploy_filtered = vinfo['deployments'][ind_press]
n_excluded = n_all - len(t_filtered)
[dataind, g_min, g_max, n_nan, n_fv,
n_grange] = index_dataset(r, vinfo['var_name'],
data_filtered, fill_value)
t_final = t_filtered[dataind]
data_final = data_filtered[dataind]
deploy_final = deploy_filtered[dataind]
t0 = pd.to_datetime(
min(t_final)).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(
max(t_final)).strftime('%Y-%m-%dT%H:%M:%S')
deploy = list(np.unique(deploy_final))
deployments = [int(dd) for dd in deploy]
if len(data_final) > 1:
[num_outliers, mean, vmin, vmax, sd, n_stats
] = cf.variable_statistics(data_final, sd_calc)
else:
mean = None
vmin = None
vmax = None
sd = None
n_stats = None
note = 'restricted stats calculation to data points where pressure is within defined ranges' \
' (average of all pressure data +/- 1 SD)'
rows.append([
m,
list(np.unique(pms)), deployments, sv, lunits, t0, t1,
fv_lst, [g_min, g_max], n_all,
[round(ipress_min, 2),
round(ipress_max,
2)], n_excluded, n_nan, n_fv, n_grange, sd_calc,
num_outliers, n_stats, mean, vmin, vmax, sd, note
])
# plot CTDMO data used for stats
psave_dir = os.path.join(plotting_sDir, array, subsite, r,
'timeseries_plots_stats')
cf.create_dir(psave_dir)
dr_data = cf.refdes_datareview_json(r)
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = cf.get_deployment_information(
dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(
pd.to_datetime(deploy_info['stop_date']))
sname = '-'.join((r, sv))
fig, ax = pf.plot_timeseries_all(t_final,
data_final,
sv,
lunits[0],
stdev=None)
ax.set_title(
(r + '\nDeployments: ' + str(sorted(deployments)) +
'\n' + t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
pf.save_fig(psave_dir, sname)
if sd_calc:
sname = '-'.join((r, sv, 'rmoutliers'))
fig, ax = pf.plot_timeseries_all(t_final,
data_final,
sv,
lunits[0],
stdev=sd_calc)
ax.set_title(
(r + '\nDeployments: ' + str(sorted(deployments)) +
'\n' + t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
pf.save_fig(psave_dir, sname)
else:
if not sd_calc:
sdcalc = None
print('\nPARAMETERS: ')
for sv, vinfo in n['vars'].items():
print(sv)
fv_lst = np.unique(vinfo['fv']).tolist()
if len(fv_lst) == 1:
fill_value = fv_lst[0]
else:
print(fv_lst)
print('No unique fill value for {}'.format(sv))
lunits = np.unique(vinfo['units']).tolist()
t = vinfo['t']
if len(t) > 1:
data = vinfo['values']
n_all = len(t)
if 'SPKIR' in r or 'presf_abc_wave_burst' in m:
if 'SPKIR' in r:
[dd_data, g_min, g_max, n_nan, n_fv,
n_grange] = index_dataset_2d(
r, 'spkir_abj_cspp_downwelling_vector',
data, fill_value)
else:
[dd_data, g_min, g_max, n_nan, n_fv,
n_grange] = index_dataset_2d(
r, 'presf_wave_burst_pressure', data,
fill_value)
t_final = t
t0 = pd.to_datetime(
min(t_final)).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(
max(t_final)).strftime('%Y-%m-%dT%H:%M:%S')
deploy_final = vinfo['deployments']
deploy = list(np.unique(deploy_final))
deployments = [int(dd) for dd in deploy]
num_outliers = []
mean = []
vmin = []
vmax = []
sd = []
n_stats = []
for i in range(len(dd_data)):
dd = data[i]
# drop nans before calculating stats
dd = dd[~np.isnan(dd)]
[
num_outliersi, meani, vmini, vmaxi, sdi,
n_statsi
] = cf.variable_statistics(dd, sd_calc)
num_outliers.append(num_outliersi)
mean.append(meani)
vmin.append(vmini)
vmax.append(vmaxi)
sd.append(sdi)
n_stats.append(n_statsi)
else:
[dataind, g_min, g_max, n_nan, n_fv,
n_grange] = index_dataset(r, vinfo['var_name'],
data, fill_value)
t_final = t[dataind]
if len(t_final) > 0:
t0 = pd.to_datetime(
min(t_final)).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(
max(t_final)).strftime('%Y-%m-%dT%H:%M:%S')
data_final = data[dataind]
# if sv == 'Dissolved Oxygen Concentration':
# xx = (data_final > 0) & (data_final < 400)
# data_final = data_final[xx]
# t_final = t_final[xx]
# if sv == 'Seawater Conductivity':
# xx = (data_final > 1) & (data_final < 400)
# data_final = data_final[xx]
# t_final = t_final[xx]
deploy_final = vinfo['deployments'][dataind]
deploy = list(np.unique(deploy_final))
deployments = [int(dd) for dd in deploy]
if len(data_final) > 1:
[
num_outliers, mean, vmin, vmax, sd,
n_stats
] = cf.variable_statistics(
data_final, sd_calc)
else:
sdcalc = None
num_outliers = None
mean = None
vmin = None
vmax = None
sd = None
n_stats = None
else:
sdcalc = None
num_outliers = None
mean = None
vmin = None
vmax = None
sd = None
n_stats = None
deployments = None
t0 = None
t1 = None
else:
sdcalc = None
num_outliers = None
mean = None
vmin = None
vmax = None
sd = None
n_stats = None
deployments = None
t0 = None
t1 = None
t_final = []
if sd_calc:
print_sd = sd_calc
else:
print_sd = sdcalc
rows.append([
m,
list(np.unique(pms)), deployments, sv, lunits, t0, t1,
fv_lst, [g_min, g_max], n_all, n_nan, n_fv, n_grange,
print_sd, num_outliers, n_stats, mean, vmin, vmax, sd
])
if len(t_final) > 0:
# plot data used for stats
psave_dir = os.path.join(
plotting_sDir, array, subsite, r,
'timeseries_reviewed_datarange')
cf.create_dir(psave_dir)
dr_data = cf.refdes_datareview_json(r)
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = cf.get_deployment_information(
dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(
pd.to_datetime(deploy_info['stop_date']))
sname = '-'.join((r, sv))
# plot hourly averages for streaming data
if 'streamed' in sci_vars_dict[list(
sci_vars_dict.keys())[0]]['ms'][0]:
sname = '-'.join((sname, 'hourlyavg'))
df = pd.DataFrame({
'dfx': t_final,
'dfy': data_final
})
dfr = df.resample('H', on='dfx').mean()
# Plot all data
fig, ax = pf.plot_timeseries_all(dfr.index,
dfr['dfy'],
sv,
lunits[0],
stdev=None)
ax.set_title((r + '\nDeployments: ' +
str(sorted(deployments)) + '\n' +
t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
pf.save_fig(psave_dir, sname)
if sd_calc:
sname = '-'.join(
(sname, 'hourlyavg_rmoutliers'))
fig, ax = pf.plot_timeseries_all(dfr.index,
dfr['dfy'],
sv,
lunits[0],
stdev=sd_calc)
ax.set_title((r + '\nDeployments: ' +
str(sorted(deployments)) + '\n' +
t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
pf.save_fig(psave_dir, sname)
elif 'SPKIR' in r:
fig, ax = pf.plot_spkir(t_final, dd_data, sv,
lunits[0])
ax.set_title((r + '\nDeployments: ' +
str(sorted(deployments)) + '\n' +
t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
pf.save_fig(psave_dir, sname)
# plot each wavelength
wavelengths = [
'412nm', '443nm', '490nm', '510nm', '555nm',
'620nm', '683nm'
]
for wvi in range(len(dd_data)):
fig, ax = pf.plot_spkir_wv(
t_final, dd_data[wvi], sv, lunits[0], wvi)
ax.set_title((r + '\nDeployments: ' +
str(sorted(deployments)) + '\n' +
t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
snamewvi = '-'.join((sname, wavelengths[wvi]))
pf.save_fig(psave_dir, snamewvi)
elif 'presf_abc_wave_burst' in m:
fig, ax = pf.plot_presf_2d(t_final, dd_data, sv,
lunits[0])
ax.set_title((r + '\nDeployments: ' +
str(sorted(deployments)) + '\n' +
t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
snamewave = '-'.join((sname, m))
pf.save_fig(psave_dir, snamewave)
else: # plot all data if not streamed
fig, ax = pf.plot_timeseries_all(t_final,
data_final,
sv,
lunits[0],
stdev=None)
ax.set_title((r + '\nDeployments: ' +
str(sorted(deployments)) + '\n' +
t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
pf.save_fig(psave_dir, sname)
if sd_calc:
sname = '-'.join((r, sv, 'rmoutliers'))
fig, ax = pf.plot_timeseries_all(t_final,
data_final,
sv,
lunits[0],
stdev=sd_calc)
ax.set_title((r + '\nDeployments: ' +
str(sorted(deployments)) + '\n' +
t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes,
color='k',
linestyle='--',
linewidth=.6)
pf.save_fig(psave_dir, sname)
fsum = pd.DataFrame(rows, columns=headers)
fsum.to_csv('{}/{}_data_ranges.csv'.format(save_dir, r), index=False)
def main(url_list, sDir, deployment_num, start_time, end_time, preferred_only,
n_std, inpercentile, zcell_size, zdbar):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list and 'ENG' not in rd and 'ADCP' not in rd:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
deployments = []
for url in url_list:
splitter = url.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
catalog_rms = '-'.join((r, splitter[-2], splitter[-1]))
if rd_check == r:
udatasets = cf.get_nc_urls([url])
for u in udatasets: # filter out collocated data files
if catalog_rms == u.split('/')[-1].split('_20')[0][15:]:
datasets.append(u)
deployments.append(
int(u.split('/')[-1].split('_')[0][-4:]))
deployments = np.unique(deployments).tolist()
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(
main_sensor, fdatasets)
for dep in deployments:
if deployment_num is not None:
if dep is not deployment_num:
print('\nskipping deployment {}'.format(dep))
continue
rdatasets = [
s for s in fdatasets_sel if 'deployment%04d' % dep in s
]
rdatasets.sort()
if len(rdatasets) > 0:
sci_vars_dict = {}
# rdatasets = rdatasets[0:2] #### for testing
for i in range(len(rdatasets)):
ds = xr.open_dataset(rdatasets[i], mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
print('\nAppending data from {}: file {} of {}'.format(
'deployment%04d' % dep, i + 1, len(rdatasets)))
array = r[0:2]
subsite = r.split('-')[0]
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'
.format(start_time, end_time))
continue
stime = start_time.strftime('%Y-%m-%d')
etime = end_time.strftime('%Y-%m-%d')
ext = stime + 'to' + etime # .join((ds0_method, ds1_method
save_dir_profile = os.path.join(
sDir, array, subsite, r, 'profile_plots',
'deployment%04d' % dep, ext)
save_dir_xsection = os.path.join(
sDir, array, subsite, r, 'xsection_plots',
'deployment%04d' % dep, ext)
else:
save_dir_profile = os.path.join(
sDir, array, subsite, r, 'profile_plots',
'deployment%04d' % dep)
save_dir_xsection = os.path.join(
sDir, array, subsite, r, 'xsection_plots',
'deployment%04d' % dep)
if len(sci_vars_dict) == 0:
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(
rdatasets[0])
sci_vars = cf.return_science_vars(stream)
if 'CTDPF' not in r:
sci_vars.append('int_ctd_pressure')
sci_vars.append('time')
sci_vars = list(np.unique(sci_vars))
# initialize the dictionary
for sci_var in sci_vars:
if sci_var == 'time':
sci_vars_dict.update({
sci_var:
dict(values=np.array([],
dtype=np.datetime64),
units=[],
fv=[])
})
else:
sci_vars_dict.update({
sci_var:
dict(values=np.array([]), units=[], fv=[])
})
# append data for the deployment into the dictionary
for s_v in sci_vars_dict.keys():
vv = ds[s_v]
try:
if vv.units not in sci_vars_dict[s_v]['units']:
sci_vars_dict[s_v]['units'].append(vv.units)
except AttributeError:
print('')
try:
if vv._FillValue not in sci_vars_dict[s_v]['fv']:
sci_vars_dict[s_v]['fv'].append(vv._FillValue)
vv_data = vv.values
try:
vv_data[
vv_data == vv.
_FillValue] = np.nan # turn fill values to nans
except ValueError:
print('')
except AttributeError:
print('')
if len(vv.dims) > 1:
print('Skipping plot: variable has >1 dimension')
else:
sci_vars_dict[s_v]['values'] = np.append(
sci_vars_dict[s_v]['values'], vv.values)
# plot after appending all data into one file
data_start = pd.to_datetime(
min(sci_vars_dict['time']['values'])).strftime(
'%Y-%m-%dT%H:%M:%S')
data_stop = pd.to_datetime(max(
sci_vars_dict['time']['values'])).strftime(
'%Y-%m-%dT%H:%M:%S')
time1 = sci_vars_dict['time']['values']
ds_lat1 = np.empty(np.shape(time1))
ds_lon1 = np.empty(np.shape(time1))
# define pressure variable
try:
pname = 'seawater_pressure'
press = sci_vars_dict[pname]
except KeyError:
pname = 'int_ctd_pressure'
press = sci_vars_dict[pname]
y1 = press['values']
try:
y_units = press['units'][0]
except IndexError:
y_units = ''
for sv in sci_vars_dict.keys():
print('')
print(sv)
if sv not in [
'seawater_pressure', 'int_ctd_pressure', 'time'
]:
z1 = sci_vars_dict[sv]['values']
fv = sci_vars_dict[sv]['fv'][0]
sv_units = sci_vars_dict[sv]['units'][0]
# Check if the array is all NaNs
if sum(np.isnan(z1)) == len(z1):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z1[z1 != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
# remove unreasonable pressure data (e.g. for surface piercing profilers)
if zdbar:
po_ind = (0 < y1) & (y1 < zdbar)
tm = time1[po_ind]
y = y1[po_ind]
z = z1[po_ind]
ds_lat = ds_lat1[po_ind]
ds_lon = ds_lon1[po_ind]
else:
tm = time1
y = y1
z = z1
ds_lat = ds_lat1
ds_lon = ds_lon1
# reject erroneous data
dtime, zpressure, ndata, lenfv, lennan, lenev, lengr, global_min, global_max, lat, lon = \
cf.reject_erroneous_data(r, sv, tm, y, z, fv, ds_lat, ds_lon)
# get rid of 0.0 data
# if sv == 'salinity':
# ind = ndata > 20
# elif sv == 'density':
# ind = ndata > 1010
# elif sv == 'conductivity':
# ind = ndata > 2
# else:
# ind = ndata > 0
# if sv == 'sci_flbbcd_chlor_units':
# ind = ndata < 7.5
# elif sv == 'sci_flbbcd_cdom_units':
# ind = ndata < 25
# else:
# ind = ndata > 0.0
if 'CTD' in r:
ind = zpressure > 0.0
else:
ind = ndata > 0.0
lenzero = np.sum(~ind)
dtime = dtime[ind]
zpressure = zpressure[ind]
ndata = ndata[ind]
if ds_lat is not None and ds_lon is not None:
lat = lat[ind]
lon = lon[ind]
else:
lat = None
lon = None
if len(dtime) > 0:
# reject time range from data portal file export
t_portal, z_portal, y_portal, lat_portal, lon_portal = \
cf.reject_timestamps_dataportal(subsite, r, dtime, zpressure, ndata, lat, lon)
print(
'removed {} data points using visual inspection of data'
.format(len(ndata) - len(z_portal)))
# create data groups
# if len(y_portal) > 0:
# columns = ['tsec', 'dbar', str(sv)]
# min_r = int(round(np.nanmin(y_portal) - zcell_size))
# max_r = int(round(np.nanmax(y_portal) + zcell_size))
# ranges = list(range(min_r, max_r, zcell_size))
#
# groups, d_groups = gt.group_by_depth_range(t_portal, y_portal, z_portal, columns, ranges)
#
# if 'scatter' in sv:
# n_std = None # to use percentile
# else:
# n_std = n_std
#
# # get percentile analysis for printing on the profile plot
# y_avg, n_avg, n_min, n_max, n0_std, n1_std, l_arr, time_ex = cf.reject_timestamps_in_groups(
# groups, d_groups, n_std, inpercentile)
"""
Plot all data
"""
if len(time1) > 0:
cf.create_dir(save_dir_profile)
cf.create_dir(save_dir_xsection)
sname = '-'.join((r, method, sv))
# sfileall = '_'.join(('all_data', sname, pd.to_datetime(time1.min()).strftime('%Y%m%d')))
# tm0 = pd.to_datetime(time1.min()).strftime('%Y-%m-%dT%H:%M:%S')
# tm1 = pd.to_datetime(time1.max()).strftime('%Y-%m-%dT%H:%M:%S')
sfileall = '_'.join(
(sname, pd.to_datetime(
t_portal.min()).strftime('%Y%m%d')))
tm0 = pd.to_datetime(t_portal.min()).strftime(
'%Y-%m-%dT%H:%M:%S')
tm1 = pd.to_datetime(t_portal.max()).strftime(
'%Y-%m-%dT%H:%M:%S')
title = ' '.join(
(deployment, refdes,
method)) + '\n' + tm0 + ' to ' + tm1
if 'SPKIR' in r:
title = title + '\nWavelength = 510 nm'
'''
profile plot
'''
xlabel = sv + " (" + sv_units + ")"
ylabel = pname + " (" + y_units + ")"
clabel = 'Time'
# fig, ax = pf.plot_profiles(z1, y1, time1, ylabel, xlabel, clabel, stdev=None)
fig, ax = pf.plot_profiles(z_portal,
y_portal,
t_portal,
ylabel,
xlabel,
clabel,
stdev=None)
ax.set_title(title, fontsize=9)
fig.tight_layout()
pf.save_fig(save_dir_profile, sfileall)
'''
xsection plot
'''
clabel = sv + " (" + sv_units + ")"
ylabel = pname + " (" + y_units + ")"
# fig, ax, bar = pf.plot_xsection(subsite, time1, y1, z1, clabel, ylabel, t_eng=None,
# m_water_depth=None, inpercentile=None, stdev=None)
fig, ax, bar = pf.plot_xsection(
subsite,
t_portal,
y_portal,
z_portal,
clabel,
ylabel,
t_eng=None,
m_water_depth=None,
inpercentile=None,
stdev=None)
if fig:
ax.set_title(title, fontsize=9)
fig.tight_layout()
pf.save_fig(save_dir_xsection, sfileall)
"""
def main(sDir, url_list, start_time, end_time, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
rms = '-'.join((r, row[ii]))
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
for fd in fdatasets:
with xr.open_dataset(fd, mask_and_scale=False) as ds:
ds = ds.swap_dims({'obs': 'time'})
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print('No data to plot for specified time range: ({} to {})'.format(start_time, end_time))
continue
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(fd)
print('\nPlotting {} {}'.format(r, deployment))
array = subsite[0:2]
save_dir = os.path.join(sDir, array, subsite, refdes, 'ts_plots')
cf.create_dir(save_dir)
tme = ds['time'].values
t0 = pd.to_datetime(tme.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(tme.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method))
filename = '-'.join(('_'.join(fname.split('_')[:-1]), 'ts', t0[:10]))
ds_vars = list(ds.data_vars.keys())
raw_vars = cf.return_raw_vars(ds_vars)
xvar = return_var(ds, raw_vars, 'salinity', 'Practical Salinity')
sal = ds[xvar].values
sal_fv = ds[xvar]._FillValue
yvar = return_var(ds, raw_vars, 'temp', 'Seawater Temperature')
temp = ds[yvar].values
temp_fv = ds[yvar]._FillValue
press = pf.pressure_var(ds, list(ds.coords.keys()))
if press is None:
press = pf.pressure_var(ds, list(ds.data_vars.keys()))
p = ds[press].values
# get rid of nans, 0.0s, fill values
sind1 = (~np.isnan(sal)) & (sal != 0.0) & (sal != sal_fv)
sal = sal[sind1]
temp = temp[sind1]
tme = tme[sind1]
p = p[sind1]
tind1 = (~np.isnan(temp)) & (temp != 0.0) & (temp != temp_fv)
sal = sal[tind1]
temp = temp[tind1]
tme = tme[tind1]
p = p[tind1]
# reject values outside global ranges:
global_min, global_max = cf.get_global_ranges(r, xvar)
if any(e is None for e in [global_min, global_max]):
sal = sal
temp = temp
tme = tme
p = p
else:
sgr_ind = cf.reject_global_ranges(sal, global_min, global_max)
sal = sal[sgr_ind]
temp = temp[sgr_ind]
tme = tme[sgr_ind]
p = p[sgr_ind]
global_min, global_max = cf.get_global_ranges(r, yvar)
if any(e is None for e in [global_min, global_max]):
sal = sal
temp = temp
tme = tme
p = p
else:
tgr_ind = cf.reject_global_ranges(temp, global_min, global_max)
sal = sal[tgr_ind]
temp = temp[tgr_ind]
tme = tme[tgr_ind]
p = p[tgr_ind]
# get rid of outliers
soind = cf.reject_outliers(sal, 5)
sal = sal[soind]
temp = temp[soind]
tme = tme[soind]
p = p[soind]
toind = cf.reject_outliers(temp, 5)
sal = sal[toind]
temp = temp[toind]
tme = tme[toind]
p = p[toind]
if len(sal) > 0: # if there are any data to plot
colors = cm.rainbow(np.linspace(0, 1, len(tme)))
# Figure out boundaries (mins and maxes)
#smin = sal.min() - (0.01 * sal.min())
#smax = sal.max() + (0.01 * sal.max())
if sal.max() - sal.min() < 0.2:
smin = sal.min() - (0.0005 * sal.min())
smax = sal.max() + (0.0005 * sal.max())
else:
smin = sal.min() - (0.001 * sal.min())
smax = sal.max() + (0.001 * sal.max())
if temp.max() - temp.min() <= 1:
tmin = temp.min() - (0.01 * temp.min())
tmax = temp.max() + (0.01 * temp.max())
elif 1 < temp.max() - temp.min() < 1.5:
tmin = temp.min() - (0.05 * temp.min())
tmax = temp.max() + (0.05 * temp.max())
else:
tmin = temp.min() - (0.1 * temp.min())
tmax = temp.max() + (0.1 * temp.max())
# Calculate how many gridcells are needed in the x and y directions and
# Create temp and sal vectors of appropriate dimensions
xdim = int(round((smax-smin)/0.1 + 1, 0))
if xdim == 1:
xdim = 2
si = np.linspace(0, xdim - 1, xdim) * 0.1 + smin
if 1.1 <= temp.max() - temp.min() < 1.7: # if the diff between min and max temp is small
ydim = int(round((tmax-tmin)/0.75 + 1, 0))
ti = np.linspace(0, ydim - 1, ydim) * 0.75 + tmin
elif temp.max() - temp.min() < 1.1:
ydim = int(round((tmax - tmin) / 0.1 + 1, 0))
ti = np.linspace(0, ydim - 1, ydim) * 0.1 + tmin
else:
ydim = int(round((tmax - tmin) + 1, 0))
ti = np.linspace(0, ydim - 1, ydim) + tmin
# Create empty grid of zeros
mdens = np.zeros((ydim, xdim))
# Loop to fill in grid with densities
for j in range(0, ydim):
for i in range(0, xdim):
mdens[j, i] = gsw.density.rho(si[i], ti[j], np.median(p)) # calculate density using median pressure value
fig, ax = pf.plot_ts(si, ti, mdens, sal, temp, colors)
ax.set_title((title + '\n' + t0 + ' - ' + t1 + '\ncolors = time (cooler: earlier)'), fontsize=9)
leg_text = ('Removed {} values (SD=5)'.format(len(ds[xvar].values) - len(sal)),)
ax.legend(leg_text, loc='best', fontsize=6)
pf.save_fig(save_dir, filename)
def main(url_list, sDir, mDir, zcell_size, zdbar, start_time, end_time, inpercentile):
"""""
URL : path to instrument data by methods
sDir : path to the directory on your machine to save plots
mDir : path to the directory on your machine to save data ranges
zcell_size : depth cell size to group data
zdbar : define depth where suspect data are identified
start_time : select start date to slice timeseries
end_time : select end date to slice timeseries
"""""
rd_list = []
ms_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
if ms not in ms_list:
ms_list.append(ms)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
main_sensor = r.split('-')[-1]
# read in the analysis file
dr_data = cf.refdes_datareview_json(r)
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
# get science variable long names from the Data Review Database
stream_sci_vars = cd.sci_var_long_names(r)
# check if the science variable long names are the same for each stream and initialize empty arrays
sci_vars_dict0 = cd.sci_var_long_names_check(stream_sci_vars)
# get the list of data files and filter out collocated instruments and other streams
datasets = []
for u in url_list:
print(u)
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
fdatasets = cf.filter_other_streams(r, ms_list, fdatasets)
# select the list of data files from the preferred dataset for each deployment
fdatasets_final = []
for ii in range(len(ps_df)):
for x in fdatasets:
if ps_df['deployment'][ii] in x and ps_df[0][ii] in x:
fdatasets_final.append(x)
# build dictionary of science data from the preferred dataset for each deployment
print('\nAppending data from files')
sci_vars_dict, y_unit, y_name, l0 = cd.append_evaluated_science_data(
sDir, ps_df, n_streams, r, fdatasets_final, sci_vars_dict0, zdbar, start_time, end_time)
# get end times of deployments
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = cf.get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# create data range output folders
save_dir_stat = os.path.join(mDir, array, subsite)
cf.create_dir(save_dir_stat)
# create plots output folder
save_fdir = os.path.join(sDir, array, subsite, r, 'data_range')
cf.create_dir(save_fdir)
stat_df = pd.DataFrame()
"""
create data ranges csv file and figures
"""
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print('\n' + vinfo['var_name'])
if len(vinfo['t']) < 1:
print('no variable data to plot')
continue
else:
sv_units = vinfo['units'][0]
fv = vinfo['fv'][0]
t = vinfo['t']
z = vinfo['values']
y = vinfo['pressure']
# Check if the array is all NaNs
if sum(np.isnan(z)) == len(z):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z[z != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
if len(y) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((vinfo['var_name'], r))
else:
sname = '-'.join((vinfo['var_name'], r, m))
"""
create data ranges for non - pressure data only
"""
if 'pressure' in vinfo['var_name']:
pass
else:
columns = ['tsec', 'dbar', str(vinfo['var_name'])]
# create depth ranges
min_r = int(round(min(y) - zcell_size))
max_r = int(round(max(y) + zcell_size))
ranges = list(range(min_r, max_r, zcell_size))
# group data by depth
groups, d_groups = gt.group_by_depth_range(t, y, z, columns, ranges)
print('writing data ranges for {}'.format(vinfo['var_name']))
stat_data = groups.describe()[vinfo['var_name']]
stat_data.insert(loc=0, column='parameter', value=sv, allow_duplicates=False)
t_deploy = deployments[0]
for i in range(len(deployments))[1:len(deployments)]:
t_deploy = '{}, {}'.format(t_deploy, deployments[i])
stat_data.insert(loc=1, column='deployments', value=t_deploy, allow_duplicates=False)
stat_df = stat_df.append(stat_data, ignore_index=False)
"""
plot full time range free from errors and suspect data
"""
clabel = sv + " (" + sv_units + ")"
ylabel = (y_name[0][0] + " (" + y_unit[0][0] + ")")
t_eng = None
m_water_depth = None
# plot non-erroneous -suspect data
fig, ax, bar = pf.plot_xsection(subsite, t, y, z, clabel, ylabel, t_eng, m_water_depth,
inpercentile, stdev=None)
title0 = 'Data colored using the upper and lower {} percentile.'.format(inpercentile)
ax.set_title(r+'\n'+title0, fontsize=9)
leg_text = ('{} % erroneous values removed after Human In the Loop review'.format(
(len(t)/l0) * 100),)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
for ii in range(len(end_times)):
ax.axvline(x=end_times[ii], color='b', linestyle='--', linewidth=.8)
ax.text(end_times[ii], min(y)-5, 'End' + str(deployments[ii]),
fontsize=6, style='italic',
bbox=dict(boxstyle='round',
ec=(0., 0.5, 0.5),
fc=(1., 1., 1.),
))
# fig.tight_layout()
sfile = '_'.join(('data_range', sname))
pf.save_fig(save_fdir, sfile)
# write stat file
stat_df.to_csv('{}/{}_data_ranges.csv'.format(save_dir_stat, r), index=True, float_format='%11.6f')
def main(sDir, url_list):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
ps_df, n_streams = cf.get_preferred_stream_info(r)
# get end times of deployments
dr_data = cf.refdes_datareview_json(r)
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# filter datasets
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
methodstream = []
for f in fdatasets:
methodstream.append('-'.join((f.split('/')[-2].split('-')[-2], f.split('/')[-2].split('-')[-1])))
for ms in np.unique(methodstream):
fdatasets_sel = [x for x in fdatasets if ms in x]
check_ms = ms.split('-')[1]
if 'recovered' in check_ms:
check_ms = check_ms.split('_recovered')
save_dir = os.path.join(sDir, array, subsite, r, 'timeseries_monthly_plot',
check_ms[0], ms.split('-')[0])
cf.create_dir(save_dir)
stream_sci_vars_dict = dict()
for x in dr_data['instrument']['data_streams']:
dr_ms = '-'.join((x['method'], x['stream_name']))
if ms == dr_ms:
stream_sci_vars_dict[dr_ms] = dict(vars=dict())
sci_vars = dict()
for y in x['stream']['parameters']:
if y['data_product_type'] == 'Science Data':
sci_vars.update({y['name']: dict(db_units=y['unit'])})
if len(sci_vars) > 0:
stream_sci_vars_dict[dr_ms]['vars'] = sci_vars
sci_vars_dict = cd.initialize_empty_arrays(stream_sci_vars_dict, ms)
print('\nAppending data from files: {}'.format(ms))
for fd in fdatasets_sel:
ds = xr.open_dataset(fd, mask_and_scale=False)
for var in list(sci_vars_dict[ms]['vars'].keys()):
sh = sci_vars_dict[ms]['vars'][var]
if ds[var].units == sh['db_units']:
if ds[var]._FillValue not in sh['fv']:
sh['fv'].append(ds[var]._FillValue)
if ds[var].units not in sh['units']:
sh['units'].append(ds[var].units)
tD = ds['time'].values
varD = ds[var].values
sh['t'] = np.append(sh['t'], tD)
sh['values'] = np.append(sh['values'], varD)
print('\nPlotting data')
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(sv)
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
t0 = pd.to_datetime(min(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
x = vinfo['t']
y = vinfo['values']
# reject NaNs
nan_ind = ~np.isnan(y)
x_nonan = x[nan_ind]
y_nonan = y[nan_ind]
# reject fill values
fv_ind = y_nonan != vinfo['fv'][0]
x_nonan_nofv = x_nonan[fv_ind]
y_nonan_nofv = y_nonan[fv_ind]
# reject extreme values
Ev_ind = cf.reject_extreme_values(y_nonan_nofv)
y_nonan_nofv_nE = y_nonan_nofv[Ev_ind]
x_nonan_nofv_nE = x_nonan_nofv[Ev_ind]
# reject values outside global ranges:
global_min, global_max = cf.get_global_ranges(r, sv)
gr_ind = cf.reject_global_ranges(y_nonan_nofv_nE, global_min, global_max)
y_nonan_nofv_nE_nogr = y_nonan_nofv_nE[gr_ind]
x_nonan_nofv_nE_nogr = x_nonan_nofv_nE[gr_ind]
title = ' '.join((r, ms.split('-')[0]))
if len(y_nonan_nofv) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((r, sv))
else:
sname = '-'.join((r, m, sv))
# 1st group by year
ygroups, gy_data = gt.group_by_timerange(x_nonan_nofv_nE_nogr, y_nonan_nofv_nE_nogr, 'A')
tn = 1
for n in range(len(ygroups)):
x_time = gy_data[n+tn].dropna(axis=0)
y_data = gy_data[n+(tn+1)].dropna(axis=0)
y_data = y_data.astype(float)
# 2nd group by month
mgroups, gm_data = gt.group_by_timerange(x_time.values, y_data.values, 'M')
x_year = x_time[0].year
print(x_year)
#
# create bins for histogram
mgroups_min = min(mgroups.describe()['DO']['min'])
mgroups_max = max(mgroups.describe()['DO']['max'])
lower_bound = int(round(mgroups_min))
upper_bound = int(round(mgroups_max + (mgroups_max / 50)))
step_bound = int(round((mgroups_max - mgroups_min) / 10))
lower_bound = int(round(global_min))
upper_bound = int(round(global_max + (global_max / 50)))
step_bound = int(round((global_max - global_min) / 10))
if step_bound == 0:
step_bound += 1
if (upper_bound - lower_bound) == step_bound:
lower_bound -= 1
upper_bound += 1
if (upper_bound - lower_bound) < step_bound:
step_bound = int(round(step_bound / 10))
bin_range = list(range(lower_bound, upper_bound, step_bound))
print(bin_range)
# create color palette
colors = color_names[:len(mgroups)]
print('1--- ', len(colors))
print(colors)
fig0, ax0 = pyplot.subplots(nrows=2, ncols=1)
# # subplot for histogram and basic statistics table
ax0[0].axis('off')
ax0[0].axis('tight')
the_table = ax0[0].table(cellText=mgroups.describe().round(2).values,
rowLabels=mgroups.describe().index.month,
rowColours=colors,
colLabels=mgroups.describe().columns.levels[1], loc='center')
the_table.set_fontsize(5)
fig, ax = pyplot.subplots(nrows=12, ncols=1, sharey=True)
for kk in list(range(0, 12)):
ax[kk].tick_params(axis='both', which='both', color='r', labelsize=7,
labelcolor='m', rotation=0, pad=0.1, length=1)
month_name = calendar.month_abbr[kk + 1]
ax[kk].set_ylabel(month_name, rotation=0, fontsize=8, color='b', labelpad=20)
if kk == 0:
ax[kk].set_title(str(x_year) + '\n ' + sv + " (" + sv_units + ")" +
' Global Range: [' + str(int(global_min)) + ',' + str(int(global_max)) + ']' +
'\n End of deployments are marked with a vertical line \n ' +
'Plotted: Data, Mean and STD (Method: 1 day' +
' rolling window calculations)',
fontsize=8)
if kk < 11:
ax[kk].tick_params(labelbottom=False)
if kk == 11:
ax[kk].set_xlabel('Days', rotation=0, fontsize=8, color='b')
tm = 1
for mt in range(len(mgroups)):
x_time = gm_data[mt+tm].dropna(axis=0)
y_data = gm_data[mt+(tm+1)].dropna(axis=0)
if len(x_time) == 0:
# ax[plt_index].tick_params(which='both', labelbottom=False, labelleft=False,
# pad=0.1, length=1)
continue
x_month = x_time[0].month
col_name = str(x_month)
series_m = pd.DataFrame(columns=[col_name], index=x_time)
series_m[col_name] = list(y_data[:])
# serie_n.plot.hist(ax=ax0[0], bins=bin_range,
# histtype='bar', color=colors[ny], stacked=True)
series_m.plot.kde(ax=ax0[0], color=colors[mt])
ax0[0].legend(fontsize=8, bbox_to_anchor=(0., 1.12, 1., .102), loc=3,
ncol=len(mgroups), mode="expand", borderaxespad=0.)
# ax0[0].set_xticks(bin_range)
ax0[0].set_xlabel('Observation Ranges' + ' (' + sv + ', ' + sv_units + ')', fontsize=8)
ax0[0].set_ylabel('Density', fontsize=8) # 'Number of Observations'
ax0[0].set_title('Kernel Density Estimates', fontsize=8)
ax0[0].tick_params(which='both', labelsize=7, pad=0.1, length=1, rotation=0)
plt_index = x_month - 1
# Plot data
series_m.plot(ax=ax[plt_index], linestyle='None', marker='.', markersize=1)
ax[plt_index].legend().set_visible(False)
ma = series_m.rolling('86400s').mean()
mstd = series_m.rolling('86400s').std()
ax[plt_index].plot(ma.index, ma[col_name].values, 'b')
ax[plt_index].fill_between(mstd.index, ma[col_name].values-3*mstd[col_name].values,
ma[col_name].values+3*mstd[col_name].values,
color='b', alpha=0.2)
# prepare the time axis parameters
mm, nod = monthrange(x_year, x_month)
datemin = datetime.date(x_year, x_month, 1)
datemax = datetime.date(x_year, x_month, nod)
ax[plt_index].set_xlim(datemin, datemax)
xlocator = mdates.DayLocator() # every day
myFmt = mdates.DateFormatter('%d')
ax[plt_index].xaxis.set_major_locator(xlocator)
ax[plt_index].xaxis.set_major_formatter(myFmt)
ax[plt_index].xaxis.set_minor_locator(pyplot.NullLocator())
ax[plt_index].xaxis.set_minor_formatter(pyplot.NullFormatter())
# data_min = min(ma.DO_n.dropna(axis=0) - 5 * mstd.DO_n.dropna(axis=0))
# 0data_max = max(ma.DO_n.dropna(axis=0) + 5 * mstd.DO_n.dropna(axis=0))
# ax[plt_index].set_ylim([data_min, data_max])
ylocator = MaxNLocator(prune='both', nbins=3)
ax[plt_index].yaxis.set_major_locator(ylocator)
if x_month != 12:
ax[plt_index].tick_params(which='both', labelbottom=False, pad=0.1, length=1)
ax[plt_index].set_xlabel(' ')
else:
ax[plt_index].tick_params(which='both', color='r', labelsize=7, labelcolor='m',
pad=0.1, length=1, rotation=0)
ax[plt_index].set_xlabel('Days', rotation=0, fontsize=8, color='b')
dep = 1
for etimes in end_times:
ax[plt_index].axvline(x=etimes, color='b', linestyle='--', linewidth=.8)
if ma[col_name].values.any():
ax[plt_index].text(etimes, max(ma[col_name].dropna(axis=0)), 'End' + str(dep),
fontsize=6, style='italic',
bbox=dict(boxstyle='round',
ec=(0., 0.5, 0.5),
fc=(1., 1., 1.),
))
else:
ax[plt_index].text(etimes, min(series_m['DO_n']), 'End' + str(dep),
fontsize=6, style='italic',
bbox=dict(boxstyle='round',
ec=(0., 0.5, 0.5),
fc=(1., 1., 1.),
))
dep += 1
tm += 1
tn += 1
# pyplot.show()
sfile = '_'.join((str(x_year), sname))
save_file = os.path.join(save_dir, sfile)
fig.savefig(str(save_file), dpi=150)
sfile = '_'.join(('Statistics', str(x_year), sname))
save_file = os.path.join(save_dir, sfile)
fig0.savefig(str(save_file), dpi=150)
def main(url_list, sDir, plot_type):
"""""
URL : path to instrument data by methods
sDir : path to the directory on your machine to save files
plot_type: folder name for a plot type
""" ""
rd_list = []
ms_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
if ms not in ms_list:
ms_list.append(ms)
'''
separate different instruments
'''
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
main_sensor = r.split('-')[-1]
ps_df, n_streams = cf.get_preferred_stream_info(r)
# read in the analysis file
dr_data = cf.refdes_datareview_json(r)
# get end times of deployments
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# get the list of data files and filter out collocated instruments and other streams chat
datasets = []
for u in url_list:
print(u)
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
fdatasets = cf.filter_other_streams(r, ms_list, fdatasets)
'''
separate the data files by methods
'''
for ms in ms_list: # np.unique(methodstream)
fdatasets_sel = [x for x in fdatasets if ms in x]
# create a folder to save figures
save_dir = os.path.join(sDir, array, subsite, r, plot_type,
ms.split('-')[0])
cf.create_dir(save_dir)
# create a dictionary for science variables from analysis file
stream_sci_vars_dict = dict()
for x in dr_data['instrument']['data_streams']:
dr_ms = '-'.join((x['method'], x['stream_name']))
if ms == dr_ms:
stream_sci_vars_dict[dr_ms] = dict(vars=dict())
sci_vars = dict()
for y in x['stream']['parameters']:
if y['data_product_type'] == 'Science Data':
sci_vars.update(
{y['name']: dict(db_units=y['unit'])})
if len(sci_vars) > 0:
stream_sci_vars_dict[dr_ms]['vars'] = sci_vars
# initialize an empty data array for science variables in dictionary
sci_vars_dict = cd.initialize_empty_arrays(stream_sci_vars_dict,
ms)
y_unit = []
y_name = []
for fd in fdatasets_sel:
ds = xr.open_dataset(fd, mask_and_scale=False)
print('\nAppending data file: {}'.format(fd.split('/')[-1]))
for var in list(sci_vars_dict[ms]['vars'].keys()):
sh = sci_vars_dict[ms]['vars'][var]
if ds[var].units == sh['db_units']:
if ds[var]._FillValue not in sh['fv']:
sh['fv'].append(ds[var]._FillValue)
if ds[var].units not in sh['units']:
sh['units'].append(ds[var].units)
# time
t = ds['time'].values
t0 = pd.to_datetime(
t.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(
t.max()).strftime('%Y-%m-%dT%H:%M:%S')
# sci variable
z = ds[var].values
sh['t'] = np.append(sh['t'], t)
sh['values'] = np.append(sh['values'], z)
# add pressure to dictionary of sci vars
if 'MOAS' in subsite:
if 'CTD' in main_sensor: # for glider CTDs, pressure is a coordinate
pressure = 'sci_water_pressure_dbar'
y = ds[pressure].values
if ds[pressure].units not in y_unit:
y_unit.append(ds[pressure].units)
if ds[pressure].long_name not in y_name:
y_name.append(ds[pressure].long_name)
else:
pressure = 'int_ctd_pressure'
y = ds[pressure].values
if ds[pressure].units not in y_unit:
y_unit.append(ds[pressure].units)
if ds[pressure].long_name not in y_name:
y_name.append(ds[pressure].long_name)
else:
pressure = pf.pressure_var(ds, ds.data_vars.keys())
y = ds[pressure].values
if ds[pressure].units not in y_unit:
y_unit.append(ds[pressure].units)
if ds[pressure].long_name not in y_name:
y_name.append(ds[pressure].long_name)
sh['pressure'] = np.append(sh['pressure'], y)
if len(y_unit) != 1:
print('pressure unit varies!')
else:
y_unit = y_unit[0]
if len(y_name) != 1:
print('pressure long name varies!')
else:
y_name = y_name[0]
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print('\nWorking on variable: {}'.format(sv))
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
fv = vinfo['fv'][0]
t0 = pd.to_datetime(min(
vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(
vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t = vinfo['t']
x = vinfo['values']
y = vinfo['pressure']
# Check if the array is all NaNs
if sum(np.isnan(x)) == len(x):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(x[x != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
# reject fill values
fv_ind = x != fv
y_nofv = y[fv_ind]
t_nofv = t[fv_ind]
c_nofv = cm.rainbow(np.linspace(0, 1, len(t[fv_ind])))
x_nofv = x[fv_ind]
print(len(x) - len(fv_ind), ' fill values')
# reject NaNs
nan_ind = ~np.isnan(x)
t_nofv_nonan = t_nofv[nan_ind]
c_nofv_nonan = c_nofv[nan_ind]
y_nofv_nonan = y_nofv[nan_ind]
x_nofv_nonan = x_nofv[nan_ind]
print(len(x) - len(nan_ind), ' NaNs')
# reject extreme values
ev_ind = cf.reject_extreme_values(x_nofv_nonan)
t_nofv_nonan_noev = t_nofv_nonan[ev_ind]
c_nofv_nonan_noev = c_nofv_nonan[ev_ind]
y_nofv_nonan_noev = y_nofv_nonan[ev_ind]
x_nofv_nonan_noev = x_nofv_nonan[ev_ind]
print(len(z) - len(ev_ind), ' Extreme Values', '|1e7|')
# reject values outside global ranges:
global_min, global_max = cf.get_global_ranges(r, sv)
# platform not in qc-table (parad_k_par)
# global_min = 0
# global_max = 2500
print('global ranges for : {}-{} {} - {}'.format(
r, sv, global_min, global_max))
if isinstance(global_min, (int, float)) and isinstance(
global_max, (int, float)):
gr_ind = cf.reject_global_ranges(
x_nofv_nonan_noev, global_min, global_max)
t_nofv_nonan_noev_nogr = t_nofv_nonan_noev[gr_ind]
y_nofv_nonan_noev_nogr = y_nofv_nonan_noev[gr_ind]
x_nofv_nonan_noev_nogr = x_nofv_nonan_noev[gr_ind]
else:
t_nofv_nonan_noev_nogr = t_nofv_nonan_noev
y_nofv_nonan_noev_nogr = y_nofv_nonan_noev
x_nofv_nonan_noev_nogr = x_nofv_nonan_noev
if len(x_nofv_nonan_noev) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((r, sv))
else:
sname = '-'.join((r, m, sv))
if sv != 'pressure':
columns = ['tsec', 'dbar', str(sv)]
bin_size = 10
min_r = int(round(min(y_nofv_nonan_noev) - bin_size))
max_r = int(round(max(y_nofv_nonan_noev) + bin_size))
ranges = list(range(min_r, max_r, bin_size))
groups, d_groups = gt.group_by_depth_range(
t_nofv_nonan_noev_nogr, y_nofv_nonan_noev_nogr,
x_nofv_nonan_noev_nogr, columns, ranges)
y_avg, n_avg, n_min, n_max, n0_std, n1_std, l_arr = [], [], [], [], [], [], []
tm = 1
for ii in range(len(groups)):
nan_ind = d_groups[ii + tm].notnull()
xtime = d_groups[ii + tm][nan_ind]
colors = cm.rainbow(np.linspace(0, 1, len(xtime)))
ypres = d_groups[ii + tm + 1][nan_ind]
nval = d_groups[ii + tm + 2][nan_ind]
tm += 2
l_arr.append(len(
nval)) # count of data to filter out small groups
y_avg.append(ypres.mean())
n_avg.append(nval.mean())
n_min.append(nval.min())
n_max.append(nval.max())
n_std = 3
n0_std.append(nval.mean() + n_std * nval.std())
n1_std.append(nval.mean() - n_std * nval.std())
# Plot all data
ylabel = y_name + " (" + y_unit + ")"
xlabel = sv + " (" + sv_units + ")"
clabel = 'Time'
fig, ax = pf.plot_profiles(x_nofv_nonan_noev_nogr,
y_nofv_nonan_noev_nogr,
t_nofv_nonan_noev_nogr,
ylabel,
xlabel,
clabel,
end_times,
deployments,
stdev=None)
title_text = ' '.join((r, ms.split('-')[-1])) + '\n' \
+ t0 + ' - ' + t1 + '\n' + str(bin_size) +\
' m average and ' + str(n_std) + ' std shown'
ax.set_title(title_text, fontsize=9)
ax.plot(n_avg, y_avg, '-k')
ax.fill_betweenx(y_avg,
n0_std,
n1_std,
color='m',
alpha=0.2)
pf.save_fig(save_dir, sname)
# Plot data with outliers removed
fig, ax = pf.plot_profiles(x_nofv_nonan_noev_nogr,
y_nofv_nonan_noev_nogr,
t_nofv_nonan_noev_nogr,
ylabel,
xlabel,
clabel,
end_times,
deployments,
stdev=5)
ax.set_title(' '.join((r, ms.split('-')[-1])) + '\n' \
+ t0 + ' - ' + t1, fontsize=9)
sfile = '_'.join((sname, 'rmoutliers'))
pf.save_fig(save_dir, sfile)
def main(sDir, url_list, preferred_only):
rd_list = []
ms_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
if ms not in ms_list:
ms_list.append(ms)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
# filter datasets
datasets = []
for u in url_list:
print(u)
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, fdatasets)
# ps_df, n_streams = cf.get_preferred_stream_info(r)
# get end times of deployments
dr_data = cf.refdes_datareview_json(r)
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# filter datasets
# datasets = []
# for u in url_list:
# splitter = u.split('/')[-2].split('-')
# rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
# if rd_check == r:
# udatasets = cf.get_nc_urls([u])
# datasets.append(udatasets)
# datasets = list(itertools.chain(*datasets))
# main_sensor = r.split('-')[-1]
# fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
# fdatasets = cf.filter_other_streams(r, ms_list, fdatasets)
methodstream = []
for f in fdatasets:
methodstream.append('-'.join((f.split('/')[-2].split('-')[-2], f.split('/')[-2].split('-')[-1])))
for ms in np.unique(methodstream):
fdatasets_sel = [x for x in fdatasets if ms in x]
save_dir = os.path.join(sDir, array, subsite, r, 'timeseries_yearly_plot', ms.split('-')[0])
cf.create_dir(save_dir)
stream_sci_vars_dict = dict()
for x in dr_data['instrument']['data_streams']:
dr_ms = '-'.join((x['method'], x['stream_name']))
if ms == dr_ms:
stream_sci_vars_dict[dr_ms] = dict(vars=dict())
sci_vars = dict()
for y in x['stream']['parameters']:
if 'light' in y['name']:
sci_vars.update({y['name']: dict(db_units=y['unit'])})
# if y['data_product_type'] == 'Science Data':
# sci_vars.update({y['name']: dict(db_units=y['unit'])})
if len(sci_vars) > 0:
stream_sci_vars_dict[dr_ms]['vars'] = sci_vars
sci_vars_dict = cd.initialize_empty_arrays(stream_sci_vars_dict, ms)
print('\nAppending data from files: {}'.format(ms))
fcount = 0
for fd in fdatasets_sel:
ds = xr.open_dataset(fd, mask_and_scale=False)
for var in list(sci_vars_dict[ms]['vars'].keys()):
sh = sci_vars_dict[ms]['vars'][var]
if ds[var].units == sh['db_units']:
if ds[var]._FillValue not in sh['fv']:
sh['fv'].append(ds[var]._FillValue)
if ds[var].units not in sh['units']:
sh['units'].append(ds[var].units)
tD = ds['time'].values
varD = ds[var].values
sh['t'] = np.append(sh['t'], tD) # put deployments time series together
if fcount == 0:
if varD.ndim > 1:
sh['values'] = np.zeros(shape=[varD.shape[0], varD.shape[1]])
sh['values'][:] = varD
else:
sh['values'] = np.append(sh['values'], varD)
else:
if varD.ndim > 1:
sh['values'] = np.vstack([sh['values'], varD])
else:
sh['values'] = np.append(sh['values'], varD)
print(fcount, var, sh['values'].shape, sh['t'].shape)
fcount += 1
print('\nPlotting data')
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(sv)
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
t0 = pd.to_datetime(min(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
x = vinfo['t']
y = vinfo['values']
if len(y) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((r, sv))
else:
sname = '-'.join((r, m, sv))
if y.ndim == 1:
num_col = 1
else:
num_col = y.shape[1]
col = [str(x) for x in list(range(1, num_col + 1))]
col.insert(0, 'time')
groups, d_groups = gt.group_by_time_frequency(x, y, col, 'A')
if len(d_groups.columns) == len(col):
time_ind = [1]
data_ind0 = [2]
data_ind1 = [len(col)]
else:
time_ind = list(range(1, len(d_groups.columns), len(col)))
data_ind0 = list(range(2, len(d_groups.columns), len(col)))
if data_ind0[-1] == d_groups.columns.values[-(len(col)+1)]:
data_ind0.insert(len(groups) - 1, len(d_groups.columns))
data_ind1 = list(range(len(col), len(d_groups.columns), len(col)))
data_ind1.insert(len(groups)-1, len(d_groups.columns))
save_file = os.path.join(save_dir, sname)
print('\ncreating images')
fig, ax = pyplot.subplots(nrows=len(groups), ncols=1)
if len(groups) == 1:
ax = [ax]
colors = color_names[:len(groups)]
images = []
for group in range(len(groups)):
nan_ind = d_groups[time_ind[group]].notnull()
xtime = d_groups[time_ind[group]][nan_ind]
ycol = list(range(data_ind0[group], data_ind1[group] + 1))
ydata = d_groups[ycol][nan_ind]
if len(xtime) != 0 and len(ydata) != 0:
n_year = xtime[0].year
print(n_year)
ydata = ydata.set_index(xtime)
if len(ydata.columns) > 1:
print('more than one col: ', len(ydata.columns))
b = fsplt.split_by_timegap(ydata, 86400)
if b:
print('gaps exist, splitting data')
for ib in (range(len(b))):
iydata = b[ib][b[ib].columns.values[0:-2]]
Y = iydata.columns.values
X = iydata.index.values
Z = iydata.values
if Z.shape[0] == 1:
X = np.repeat(X[0], len(Y), axis=0)
df = pd.DataFrame(dict(a=list(X), b=list(Y), c=list(Z[0])))
images.append(ax[group].scatter(df['a'].values, df['b'].values, c=df['c'].values, cmap='Blues', s=1))
else:
Z = Z.T
x, y = np.meshgrid(X, Y)
images.append(ax[group].contourf(x, y, Z, alpha=0.7, cmap='Blues'))
else:
print('no gaps exist, not splitting data')
iydata = ydata[ydata.columns.values[0:-2]]
Y = iydata.columns.values
X = iydata.index.values
Z = iydata.values
if Z.shape[0] == 1:
X = np.repeat(X[0], len(Y), axis=0)
df = pd.DataFrame(dict(a=list(X), b=list(Y), c=list(Z[0])))
images.append(
ax[group].scatter(df['a'].values, df['b'].values, c=df['c'].values,
cmap='Blues', s=1))
elif Z.shape[1] < 2:
Y = np.repeat(Y[0], len(X), axis=1)
df = pd.DataFrame(dict(a=list(X), b=list(Y), c=list(Z[0])))
images.append(
ax[group].scatter(df['a'].values, df['b'].values, c=df['c'].values,
cmap='Blues', s=1))
else:
Z = Z.T
x, y = np.meshgrid(X, Y)
im = ax[group].contourf(x, y, Z, alpha=0.7, cmap='Blues') #pyplot.cm.jet
images.append(ax[group].contourf(x, y, Z, alpha=0.7, cmap='Blues'))
else:
print('with one column:', len(ydata.columns))
ax[group].contourf(x, y, Z, alpha=0.7, cmap='Blues')(ydata.plot(ax=ax[group],
linestyle='None',
marker='.',
markersize=0.5,
color=colors[group]))
ax[group].legend().set_visible(False)
# plot Mean and Standard deviation
ma = ydata.rolling('86400s').mean()
mstd = ydata.rolling('86400s').std()
m_mstd_min = ma[ycol].values - 2 * mstd[ycol].values
m_mstd_max = ma[ycol].values + 2 * mstd[ycol].values
ax[group].plot(ma.index.values, ma[ycol].values, 'k', linewidth=0.15)
ax[group].fill_between(mstd.index.values, m_mstd_min, m_mstd_max, color='b', alpha=0.2)
# flag deployments end-time for reference
ymin, ymax = ax[group].get_ylim()
dep = 1
for etimes in end_times:
if etimes.year == n_year:
ax[group].axvline(x=etimes, color='b', linestyle='--', linewidth=.6)
ax[group].text(etimes, ymin, 'End' + str(dep), fontsize=6, style='italic',
bbox=dict(boxstyle='round', ec=(0., 0.5, 0.5), fc=(1., 1., 1.)))
dep += 1
# prepare the time axis parameters
datemin = datetime.date(n_year, 1, 1)
datemax = datetime.date(n_year, 12, 31)
ax[group].set_xlim(datemin, datemax)
xlocator = mdates.MonthLocator() # every month
myFmt = mdates.DateFormatter('%m')
ax[group].xaxis.set_minor_locator(xlocator)
ax[group].xaxis.set_major_formatter(myFmt)
# prepare the y axis parameters
ax[group].set_ylabel(n_year, rotation=0, fontsize=8, color='b', labelpad=20)
ax[group].yaxis.set_label_position("right")
ylocator = MaxNLocator(prune='both', nbins=3)
ax[group].yaxis.set_major_locator(ylocator)
ax[group].yaxis.set_ticklabels([]) #range(1, len(col), 1)
# format figure
ax[group].tick_params(axis='both', color='r', labelsize=7, labelcolor='m')
if group == 0:
ax[group].set_title(sv + '( ' + sv_units + ')', fontsize=8)
if group < len(groups) - 1:
ax[group].tick_params(which='both', pad=0.1, length=1, labelbottom=False)
ax[group].set_xlabel(' ')
else:
ax[group].tick_params(which='both', color='r', labelsize=7, labelcolor='m',
pad=0.1, length=1, rotation=0)
ax[group].set_xlabel('Months', rotation=0, fontsize=8, color='b')
vmin = min(image.get_array().min() for image in images)
vmax = max(image.get_array().max() for image in images)
norm = mcolors.Normalize(vmin=vmin, vmax=vmax)
for im in images:
im.set_norm(norm)
fig.colorbar(images[0], ax=ax, orientation='horizontal', fraction=.1, spacing='proportional')
fig.savefig(str(save_file), dpi=150)
pyplot.close()
def main(sDir, url_list):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
ps_df, n_streams = cf.get_preferred_stream_info(r)
# get end times of deployments
dr_data = cf.refdes_datareview_json(r)
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# filter datasets
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
methodstream = []
for f in fdatasets:
methodstream.append('-'.join((f.split('/')[-2].split('-')[-2],
f.split('/')[-2].split('-')[-1])))
for ms in np.unique(methodstream):
fdatasets_sel = [x for x in fdatasets if ms in x]
save_dir = os.path.join(sDir, array, subsite, r,
'timeseries_daily_plots',
ms.split('-')[0])
cf.create_dir(save_dir)
stream_sci_vars_dict = dict()
for x in dr_data['instrument']['data_streams']:
dr_ms = '-'.join((x['method'], x['stream_name']))
if ms == dr_ms:
stream_sci_vars_dict[dr_ms] = dict(vars=dict())
sci_vars = dict()
for y in x['stream']['parameters']:
if y['data_product_type'] == 'Science Data':
sci_vars.update(
{y['name']: dict(db_units=y['unit'])})
if len(sci_vars) > 0:
stream_sci_vars_dict[dr_ms]['vars'] = sci_vars
sci_vars_dict = cd.initialize_empty_arrays(stream_sci_vars_dict,
ms)
print('\nAppending data from files: {}'.format(ms))
for fd in fdatasets_sel:
ds = xr.open_dataset(fd, mask_and_scale=False)
for var in list(sci_vars_dict[ms]['vars'].keys()):
sh = sci_vars_dict[ms]['vars'][var]
if ds[var].units == sh['db_units']:
if ds[var]._FillValue not in sh['fv']:
sh['fv'].append(ds[var]._FillValue)
if ds[var].units not in sh['units']:
sh['units'].append(ds[var].units)
tD = ds['time'].values
varD = ds[var].values
sh['t'] = np.append(sh['t'], tD)
sh['values'] = np.append(sh['values'], varD)
print('\nPlotting data')
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(sv)
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
t0 = pd.to_datetime(min(
vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(
vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
x = vinfo['t']
y = vinfo['values']
# reject NaNs
nan_ind = ~np.isnan(y)
x_nonan = x[nan_ind]
y_nonan = y[nan_ind]
# reject fill values
fv_ind = y_nonan != vinfo['fv'][0]
x_nonan_nofv = x_nonan[fv_ind]
y_nonan_nofv = y_nonan[fv_ind]
# reject extreme values
Ev_ind = cf.reject_extreme_values(y_nonan_nofv)
y_nonan_nofv_nE = y_nonan_nofv[Ev_ind]
x_nonan_nofv_nE = x_nonan_nofv[Ev_ind]
# reject values outside global ranges:
global_min, global_max = cf.get_global_ranges(r, sv)
gr_ind = cf.reject_global_ranges(
y_nonan_nofv_nE, global_min, global_max)
y_nonan_nofv_nE_nogr = y_nonan_nofv_nE[gr_ind]
x_nonan_nofv_nE_nogr = x_nonan_nofv_nE[gr_ind]
if len(y_nonan_nofv) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((r, sv))
else:
sname = '-'.join((r, m, sv))
# 1st group by year
ygroups, gy_data = gt.group_by_timerange(
x_nonan_nofv_nE_nogr, y_nonan_nofv_nE_nogr,
'A')
tn = 1
for n in range(len(ygroups)):
x_time = gy_data[n + tn].dropna(axis=0)
y_data = gy_data[n + (tn + 1)].dropna(axis=0)
# 2nd group by month
mgroups, gm_data = gt.group_by_timerange(
x_time.values, y_data.values, 'M')
if len(x_time) == 0:
continue
td = 1
for jj in range(len(mgroups)):
x_time = gm_data[jj + td].dropna(axis=0)
y_data = gm_data[jj +
(td + 1)].dropna(axis=0)
if len(x_time) == 0:
continue
# 3rd group by day
dgroups, gd_data = gt.group_by_timerange(
x_time.values, y_data.values, 'D')
x_year = x_time[0].year
x_month = x_time[0].month
month_name = calendar.month_abbr[x_month]
print(x_year, x_month)
sfile = '_'.join(
(str(x_year), str(x_month), sname))
# prepare plot layout
fig, ax = pyplot.subplots(nrows=7,
ncols=5,
sharey=True)
title_in = month_name + '-' + str(x_year) + \
' calendar days \n Parameter: ' + \
sv + " (" + sv_units + ")"
ax[0][2].text(0.5,
1.5,
title_in,
horizontalalignment='center',
fontsize=8,
transform=ax[0][2].transAxes)
num_i = 0
day_i = {}
for kk in list(range(0, 7)):
for ff in list(range(0, 5)):
num_i += 1
day_i[num_i] = [kk, ff]
ax[kk][ff].tick_params(
axis='both',
which='both',
color='r',
labelsize=7,
labelcolor='m',
rotation=0)
ax[kk][ff].text(
0.1,
0.75,
str(num_i),
horizontalalignment='center',
fontsize=7,
transform=ax[kk][ff].transAxes,
bbox=dict(
boxstyle="round",
ec=(0., 0.5, 0.5),
fc=(1., 1., 1.),
))
if kk is not 6:
ax[kk][ff].tick_params(
labelbottom=False)
if ff is not 0:
ax[kk][ff].tick_params(
labelright=False)
if kk is 6 and ff is 0:
ax[kk][ff].set_xlabel(
'Hours',
rotation=0,
fontsize=8,
color='b')
if kk is 6 and ff in list(
range(1, 5)):
fig.delaxes(ax[kk][ff])
tm = 1
for mt in range(len(dgroups)):
x_time = gd_data[mt +
tm].dropna(axis=0)
y_DO = gd_data[mt +
(tm + 1)].dropna(axis=0)
series_m = pd.DataFrame(
columns=['DO_n'], index=x_time)
series_m['DO_n'] = list(y_DO[:])
if len(x_time) == 0:
continue
x_day = x_time[0].day
print(x_time[0].year, x_time[0].month,
x_day)
i0 = day_i[x_day][0]
i1 = day_i[x_day][1]
# Plot data
series_m.plot(ax=ax[i0][i1],
linestyle='None',
marker='.',
markersize=1)
ax[i0][i1].legend().set_visible(False)
ma = series_m.rolling('3600s').mean()
mstd = series_m.rolling('3600s').std()
ax[i0][i1].plot(ma.index,
ma.DO_n,
'b',
linewidth=0.25)
ax[i0][i1].fill_between(
mstd.index,
ma.DO_n - 3 * mstd.DO_n,
ma.DO_n + 3 * mstd.DO_n,
color='b',
alpha=0.2)
# prepare the time axis parameters
datemin = datetime.datetime(
x_year, x_month, x_day, 0)
datemax = datetime.datetime(
x_year, x_month, x_day, 23)
ax[i0][i1].set_xlim(datemin, datemax)
xLocator = mdates.HourLocator(
interval=4) # every hour
myFmt = mdates.DateFormatter('%H')
ax[i0][i1].xaxis.set_minor_locator(
xLocator)
ax[i0][i1].xaxis.set_minor_formatter(
myFmt)
ax[i0][i1].xaxis.set_major_locator(
pyplot.NullLocator())
ax[i0][i1].xaxis.set_major_formatter(
pyplot.NullFormatter())
yLocator = MaxNLocator(prune='both',
nbins=3)
ax[i0][i1].yaxis.set_major_locator(
yLocator)
if x_day is not 31:
ax[i0][i1].tick_params(
labelbottom=False)
ax[i0][i1].set_xlabel(' ')
else:
ax[i0][i1].tick_params(
which='both',
color='r',
labelsize=7,
labelcolor='m',
length=0.1,
pad=0.1)
ax[i0][i1].set_xlabel('Hours',
rotation=0,
fontsize=8,
color='b')
ymin, ymax = ax[i0][i1].get_ylim()
dep = 1
for etimes in end_times:
ax[i0][i1].axvline(x=etimes,
color='b',
linestyle='--',
linewidth=.6)
ax[i0][i1].text(
etimes,
ymin + 50,
str(dep),
fontsize=6,
style='italic',
bbox=dict(
boxstyle="round",
ec=(0., 0.5, 0.5),
fc=(1., 1., 1.),
))
dep += 1
tm += 1
td += 1
pf.save_fig(save_dir, sfile)
tn += 1
def main(url_list, sDir, plot_type):
"""""
URL : path to instrument data by methods
sDir : path to the directory on your machine to save files
plot_type: folder name for a plot type
""" ""
rd_list = []
ms_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
if ms not in ms_list:
ms_list.append(ms)
'''
separate different instruments
'''
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
main_sensor = r.split('-')[-1]
ps_df, n_streams = cf.get_preferred_stream_info(r)
# read in the analysis file
dr_data = cf.refdes_datareview_json(r)
# get preferred stream
ps_df, n_streams = cf.get_preferred_stream_info(r)
# get end times of deployments
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = cf.get_deployment_information(
dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# get the list of data files and filter out collocated instruments and other streams
datasets = []
for u in url_list:
print(u)
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
fdatasets = cf.filter_other_streams(r, ms_list, fdatasets)
'''
separate data files by methods
'''
for ms in ms_list:
fdatasets_sel = [x for x in fdatasets if ms in x]
# create a folder to save figures
save_dir = os.path.join(sDir, array, subsite, r, plot_type,
ms.split('-')[0])
cf.create_dir(save_dir)
# create a dictionary for science variables from analysis file
stream_sci_vars_dict = dict()
for x in dr_data['instrument']['data_streams']:
dr_ms = '-'.join((x['method'], x['stream_name']))
if ms == dr_ms:
stream_sci_vars_dict[dr_ms] = dict(vars=dict())
sci_vars = dict()
for y in x['stream']['parameters']:
if y['data_product_type'] == 'Science Data':
sci_vars.update(
{y['name']: dict(db_units=y['unit'])})
if len(sci_vars) > 0:
stream_sci_vars_dict[dr_ms]['vars'] = sci_vars
# initialize an empty data array for science variables in dictionary
sci_vars_dict = cd.initialize_empty_arrays(stream_sci_vars_dict,
ms)
print('\nAppending data from files: {}'.format(ms))
y_unit = []
y_name = []
for fd in fdatasets_sel:
ds = xr.open_dataset(fd, mask_and_scale=False)
print('\nAppending data file: {}'.format(fd.split('/')[-1]))
for var in list(sci_vars_dict[ms]['vars'].keys()):
sh = sci_vars_dict[ms]['vars'][var]
if ds[var].units == sh['db_units']:
if ds[var]._FillValue not in sh['fv']:
sh['fv'].append(ds[var]._FillValue)
if ds[var].units not in sh['units']:
sh['units'].append(ds[var].units)
# time
t = ds['time'].values
t0 = pd.to_datetime(
t.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(
t.max()).strftime('%Y-%m-%dT%H:%M:%S')
# sci variable
z = ds[var].values
sh['t'] = np.append(sh['t'], t)
sh['values'] = np.append(sh['values'], z)
# add pressure to dictionary of sci vars
if 'MOAS' in subsite:
if 'CTD' in main_sensor: # for glider CTDs, pressure is a coordinate
pressure = 'sci_water_pressure_dbar'
y = ds[pressure].values
if ds[pressure].units not in y_unit:
y_unit.append(ds[pressure].units)
if ds[pressure].long_name not in y_name:
y_name.append(ds[pressure].long_name)
else:
pressure = 'int_ctd_pressure'
y = ds[pressure].values
if ds[pressure].units not in y_unit:
y_unit.append(ds[pressure].units)
if ds[pressure].long_name not in y_name:
y_name.append(ds[pressure].long_name)
else:
pressure = pf.pressure_var(ds, ds.data_vars.keys())
y = ds[pressure].values
sh['pressure'] = np.append(sh['pressure'], y)
try:
ds[pressure].units
if ds[pressure].units not in y_unit:
y_unit.append(ds[pressure].units)
except AttributeError:
print('pressure attributes missing units')
if 'pressure unit missing' not in y_unit:
y_unit.append('pressure unit missing')
try:
ds[pressure].long_name
if ds[pressure].long_name not in y_name:
y_name.append(ds[pressure].long_name)
except AttributeError:
print('pressure attributes missing long_name')
if 'pressure long name missing' not in y_name:
y_name.append('pressure long name missing')
# create a csv file with diagnostic results:
if len(y_unit) != 1:
print('pressure unit varies')
if 'dbar' in y_unit:
y_unit = 'dbar'
print(y_unit)
else:
y_unit = y_unit[0]
if len(y_name) != 1:
print('pressure long name varies')
if 'Seawater Pressure' in y_name:
y_name = 'Seawater Pressure'
print(y_name)
else:
y_name = y_name[0]
# create a folder to save variables statistics
mDir = '/Users/leila/Documents/NSFEduSupport/github/data-review-tools/data_review/final_stats'
save_dir_stat = os.path.join(mDir, array, subsite)
cf.create_dir(save_dir_stat)
stat_df = pd.DataFrame()
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(sv)
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
fv = vinfo['fv'][0]
t0 = pd.to_datetime(min(
vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(
vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t = vinfo['t']
z = vinfo['values']
y = vinfo['pressure']
title = ' '.join((r, ms))
# Check if the array is all NaNs
if sum(np.isnan(z)) == len(z):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z[z != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
# reject fill values
fv_ind = z != fv
y_nofv = y[fv_ind]
t_nofv = t[fv_ind]
z_nofv = z[fv_ind]
print(len(z) - len(fv_ind), ' fill values')
# reject NaNs
nan_ind = ~np.isnan(z_nofv)
t_nofv_nonan = t_nofv[nan_ind]
y_nofv_nonan = y_nofv[nan_ind]
z_nofv_nonan = z_nofv[nan_ind]
print(len(z) - len(nan_ind), ' NaNs')
# reject extreme values
ev_ind = cf.reject_extreme_values(z_nofv_nonan)
t_nofv_nonan_noev = t_nofv_nonan[ev_ind]
y_nofv_nonan_noev = y_nofv_nonan[ev_ind]
z_nofv_nonan_noev = z_nofv_nonan[ev_ind]
print(
len(z) - len(ev_ind), ' Extreme Values',
'|1e7|')
# reject values outside global ranges:
global_min, global_max = cf.get_global_ranges(
r, sv)
# platform not in qc-table (parad_k_par)
# global_min = 0
# global_max = 2500
print('global ranges for : {}-{} {} - {}'.format(
r, sv, global_min, global_max))
if isinstance(global_min,
(int, float)) and isinstance(
global_max, (int, float)):
gr_ind = cf.reject_global_ranges(
z_nofv_nonan_noev, global_min, global_max)
t_nofv_nonan_noev_nogr = t_nofv_nonan_noev[
gr_ind]
y_nofv_nonan_noev_nogr = y_nofv_nonan_noev[
gr_ind]
z_nofv_nonan_noev_nogr = z_nofv_nonan_noev[
gr_ind]
else:
t_nofv_nonan_noev_nogr = t_nofv_nonan_noev
y_nofv_nonan_noev_nogr = y_nofv_nonan_noev
z_nofv_nonan_noev_nogr = z_nofv_nonan_noev
if len(z_nofv_nonan_noev) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((r, sv))
else:
sname = '-'.join((r, m, sv))
# group by depth range
sname = '_'.join((sname, sv_units))
# if sv != 'pressure':
# columns = ['tsec', 'dbar', str(sv)]
#
# # select depth bin size for the data group function
# bin_size = 10
# min_r = int(round(min(y_nofv_nonan_noev) - bin_size))
# max_r = int(round(max(y_nofv_nonan_noev) + bin_size))
# ranges = list(range(min_r, max_r, bin_size))
# groups, d_groups = gt.group_by_depth_range(t_nofv_nonan_noev_nogr, y_nofv_nonan_noev_nogr,
# z_nofv_nonan_noev_nogr, columns, ranges)
#
# if (ms.split('-')[0]) == (ps_df[0].values[0].split('-')[0]):
# if 'pressure' not in sv:
# print('final_stats_{}-{}-{}-{}'.format(r,
# ms.split('-')[0],
# ps_df[0].values[0].split('-')[0],
# sv))
# stat_data = groups.describe()[sv]
# stat_data.insert(loc=0, column='parameter', value=sv, allow_duplicates=False)
# stat_df = stat_df.append(stat_data)
# if sv == 'optical_backscatter':
# less_ind = z_nofv_nonan_noev < 0.0004
# print(sv, ' < 0.0004', len(less_ind))
# more_ind = z_nofv_nonan_noev > 0.01
# print(sv, ' > 0.01', len(more_ind))
# Plot all data
clabel = sv + " (" + sv_units + ")"
ylabel = y_name + " (" + y_unit + ")"
fig, ax = pf.plot_xsection(subsite,
t_nofv_nonan_noev,
y_nofv_nonan_noev,
z_nofv_nonan_noev,
clabel,
ylabel,
stdev=None)
ax.set_title((title + '\n' + t0 + ' - ' + t1),
fontsize=9)
pf.save_fig(save_dir, sname)
# Plot data with outliers removed
fig, ax = pf.plot_xsection(subsite,
t_nofv_nonan_noev_nogr,
y_nofv_nonan_noev_nogr,
z_nofv_nonan_noev_nogr,
clabel,
ylabel,
stdev=5)
ax.set_title((title + '\n' + t0 + ' - ' + t1),
fontsize=9)
sfile = '_'.join((sname, 'rmoutliers'))
pf.save_fig(save_dir, sfile)
# plot data with excluded time range removed
dr = pd.read_csv(
'https://datareview.marine.rutgers.edu/notes/export'
)
drn = dr.loc[dr.type == 'exclusion']
if len(drn) != 0:
subsite_node = '-'.join((subsite, r.split('-')[1]))
drne = drn.loc[drn.reference_designator.isin(
[subsite, subsite_node, r])]
t_ex = t_nofv_nonan_noev_nogr
y_ex = y_nofv_nonan_noev_nogr
z_ex = z_nofv_nonan_noev_nogr
for i, row in drne.iterrows():
sdate = cf.format_dates(row.start_date)
edate = cf.format_dates(row.end_date)
ts = np.datetime64(sdate)
te = np.datetime64(edate)
ind = np.where((t_ex < ts) | (t_ex > te), True,
False)
if len(ind) != 0:
t_ex = t_ex[ind]
z_ex = z_ex[ind]
y_ex = y_ex[ind]
fig, ax = pf.plot_xsection(subsite,
t_ex,
y_ex,
z_ex,
clabel,
ylabel,
stdev=None)
ax.set_title((title + '\n' + t0 + ' - ' + t1),
fontsize=9)
sfile = '_'.join((sname, 'rmsuspectdata'))
pf.save_fig(save_dir, sfile)
def main(sDir, url_list, preferred_only, name_list):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
if preferred_only == 'yes':
ps_df, n_streams = cf.get_preferred_stream_info(r)
fdatasets = []
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, fdatasets)
vars_df = compare_variable_attributes(fdatasets, r, name_list, sDir)
if len(np.unique(vars_df.index.values)) == 1 and len(vars_df['var_id']) == len(name_list)+1:
print('Pass: variables exist in all files')
else:
print('Fail: variables differ in between files')
# all_in_one_df = append_data(vars_df, fdatasets)
vars_df.insert(loc=len(vars_df.columns), column='values', value=vars_df['var_id'])
vars_df.insert(loc=len(vars_df.columns), column='t0', value=vars_df['var_id'])
vars_df.insert(loc=len(vars_df.columns), column='t1', value=vars_df['var_id'])
vars_df.insert(loc=len(vars_df.columns), column='deployments', value=vars_df['var_id'])
vars_df.insert(loc=len(vars_df.columns), column='preferred_methods_streams', value=vars_df['var_id'])
vars_df.insert(loc=len(vars_df.columns), column='common_stream_name', value=vars_df['var_id'])
vars_df.index = vars_df['var_id']
add_to_df = pd.DataFrame()
for ii in range(len(fdatasets)): # [4, 5]
print('\n', fdatasets[ii].split('/')[-1])
ds = xr.open_dataset(fdatasets[ii], mask_and_scale=False)
var_ms = '-'.join((ds.collection_method, ds.stream))
var_s = ds.stream
var_d = ds['deployment'].values
time_d = ds['time'].values
for vv in range(len(vars_df['var_id'])):
print(vars_df['var_id'].values[vv])
if ii == 0: #if ii == 4
vD = vars_df[vars_df['values'].values == vars_df['var_id'].values[vv]]
data_v = ds[vD['var_id'][0]].values
if len(np.unique(data_v)) == 1:
data_v = np.unique(data_v)
vD.at[vD.index.values[0], 'values'] = data_v
vD.at[vD.index.values[0], 'deployments'] = np.unique(var_d)
vD.at[vD.index.values[0], 't0'] = time_d[0]
vD.at[vD.index.values[0], 't1'] = time_d[len(time_d)-1]
vD.at[vD.index.values[0], 'preferred_methods_streams'] = np.unique(var_ms)
vD.at[vD.index.values[0], 'common_stream_name'] = np.unique(var_s)
else:
vD = add_to_df[add_to_df['var_id'].values == vars_df['var_id'].values[vv]]
data_v = ds[vD['var_id'][0]].values
if len(np.unique(data_v)) == 1:
data_v = np.unique(data_v)
vD.at[vD.index.values[0], 'values'] = np.append(vD['values'].values[0], data_v)
vD.at[vD.index.values[0], 'deployments'] = np.unique(np.append(vD['deployments'].values[0], var_d))
vD.at[vD.index.values[0], 't0'] = np.unique(np.append(vD['t0'].values[0],time_d[0]))
vD.at[vD.index.values[0], 't1'] = np.unique(np.append(vD['t1'].values[0],time_d[len(time_d) - 1]))
vD.at[vD.index.values[0], 'preferred_methods_streams'] = np.unique(np.append(
vD['preferred_methods_streams'].values[0], var_ms))
vD.at[vD.index.values[0], 'common_stream_name'] = np.unique(np.append(vD['common_stream_name'].values[0], var_s))
add_to_df = add_to_df.append(vD)
all_in_one_df = add_to_df[len(add_to_df) - len(vars_df['var_id']):len(add_to_df)]
df_roll = all_in_one_df[all_in_one_df.var_id.str.contains('roll') == True]
df_pitch = all_in_one_df[all_in_one_df.var_id.str.contains('pitch') == True]
# define indices to select data with pitch or roll less than 20 degrees
ind = np.logical_and(df_roll['values'].values[0] < 200, df_pitch['values'].values[0] < 200)
df_F = pd.DataFrame()
for keyword in ['eastward', 'northward', 'upward', 'pressure']: #
df_k = all_in_one_df[all_in_one_df.var_id.str.contains(keyword) == True]
print(keyword, len(df_k['t0'].values), len(df_k['t1'].values))
if len(df_k['t1'].values) > 1:
df_k.at[df_k.index.values[0], 't1'] = max(df_k['t1'].values[0])
if len(df_k['t0'].values) > 1:
df_k.at[df_k.index.values[0], 't0'] = min(df_k['t0'].values[0])
u = df_k['values'].values[0]
uu = u[ind]
uu_fv = float(df_k['fill_values'].values[0])
uu_id = df_k['var_id'].values[0]
u_wo_err, u_err = reject_err_data_1_dims(uu, uu_fv, r, uu_id, n=None)
u_err.insert(loc=0, column='n_all', value=len(u))
u_err.insert(loc=1, column='n_pitchroll_err', value=len(u) - len(uu))
u_err.insert(loc=len(u_err.columns), column='mean', value=np.nanmean(u_wo_err))
u_err.insert(loc=len(u_err.columns), column='min', value=np.nanmin(u_wo_err))
u_err.insert(loc=len(u_err.columns), column='max', value=np.nanmax(u_wo_err))
u_err.insert(loc=len(u_err.columns), column='stdev', value=np.nanstd(u_wo_err))
for name in u_err.columns:
df_k.insert(loc=len(df_k.columns), column=name, value=u_err[name].values)
df_F = df_F.append(df_k)
columns = ['common_stream_name', 'preferred_methods_streams', 'deployments', 'long_name', 'units',
't0', 't1', 'fill_values', 'global_ranges', 'n_all', 'n_pitchroll_err', 'n_nans', 'n_fillvalues',
'n_extremvalues', 'n_grange', 'define_stdev', 'n_outliers', 'n_stats', 'mean', 'min', 'max', 'stdev']
df_F.to_csv(sDir+array+'/'+subsite+'/'+r+'_data_ranges.csv', columns=columns, index=False)
def main(url_list, sDir, plot_type, deployment_num, start_time, end_time, method_num, zdbar, n_std, inpercentile, zcell_size):
for i, u in enumerate(url_list):
print('\nUrl {} of {}: {}'.format(i + 1, len(url_list), u))
elements = u.split('/')[-2].split('-')
r = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = u.split(r + '-')[1].split('/')[0]
subsite = r.split('-')[0]
array = subsite[0:2]
main_sensor = r.split('-')[-1]
# read URL to get data
datasets = cf.get_nc_urls([u])
datasets_sel = cf.filter_collocated_instruments(main_sensor, datasets)
# get sci data review list
dr_data = cf.refdes_datareview_json(r)
ps_df, n_streams = cf.get_preferred_stream_info(r)
# get end times of deployments
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = cf.get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# create a dictionary for science variables from analysis file
stream_sci_vars_dict = dict()
for x in dr_data['instrument']['data_streams']:
dr_ms = '-'.join((x['method'], x['stream_name']))
if ms == dr_ms:
stream_sci_vars_dict[dr_ms] = dict(vars=dict())
sci_vars = dict()
for y in x['stream']['parameters']:
if y['data_product_type'] == 'Science Data':
sci_vars.update({y['name']: dict(db_units=y['unit'])})
if len(sci_vars) > 0:
stream_sci_vars_dict[dr_ms]['vars'] = sci_vars
for ii, d in enumerate(datasets_sel):
part_d = d.split('/')[-1]
print('\nDataset {} of {}: {}'.format(ii + 1, len(datasets_sel), part_d))
with xr.open_dataset(d, mask_and_scale=False) as ds:
ds = ds.swap_dims({'obs': 'time'})
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(d)
if method_num is not None:
if method != method_num:
print(method_num, method)
continue
if deployment_num is not None:
if int(deployment.split('0')[-1]) is not deployment_num:
print(type(int(deployment.split('0')[-1])), type(deployment_num))
continue
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print('No data to plot for specified time range: ({} to {})'.format(start_time, end_time))
continue
stime = start_time.strftime('%Y-%m-%d')
etime = end_time.strftime('%Y-%m-%d')
ext = stime + 'to' + etime # .join((ds0_method, ds1_method
save_dir = os.path.join(sDir, array, subsite, refdes, plot_type, ms.split('-')[0], deployment, ext)
else:
save_dir = os.path.join(sDir, array, subsite, refdes, plot_type, ms.split('-')[0], deployment)
cf.create_dir(save_dir)
texclude_dir = os.path.join(sDir, array, subsite, refdes, 'time_to_exclude')
cf.create_dir(texclude_dir)
# initialize an empty data array for science variables in dictionary
sci_vars_dict = cd.initialize_empty_arrays(stream_sci_vars_dict, ms)
for var in list(sci_vars_dict[ms]['vars'].keys()):
sh = sci_vars_dict[ms]['vars'][var]
if ds[var].units == sh['db_units']:
if ds[var]._FillValue not in sh['fv']:
sh['fv'].append(ds[var]._FillValue)
if ds[var].units not in sh['units']:
sh['units'].append(ds[var].units)
sh['t'] = np.append(sh['t'], ds['time'].values) # t = ds['time'].values
sh['values'] = np.append(sh['values'], ds[var].values) # z = ds[var].values
y, y_unit, y_name = cf.add_pressure_to_dictionary_of_sci_vars(ds)
sh['pressure'] = np.append(sh['pressure'], y)
stat_data = pd.DataFrame(columns=['deployments', 'time_to_exclude'])
file_exclude = '{}/{}_{}_{}_excluded_timestamps.csv'.format(texclude_dir,
deployment, refdes, method)
stat_data.to_csv(file_exclude, index=True)
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(sv)
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
fv = vinfo['fv'][0]
t0 = pd.to_datetime(min(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
colors = cm.rainbow(np.linspace(0, 1, len(vinfo['t'])))
t = vinfo['t']
z = vinfo['values']
y = vinfo['pressure']
# Check if the array is all NaNs
if sum(np.isnan(z)) == len(z):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z[z != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
# reject erroneous data
dtime, zpressure, ndata, lenfv, lennan, lenev, lengr, global_min, global_max = \
cf.reject_erroneous_data(r, sv, t, y, z, fv)
# create data groups
columns = ['tsec', 'dbar', str(sv)]
min_r = int(round(min(zpressure) - zcell_size))
max_r = int(round(max(zpressure) + zcell_size))
ranges = list(range(min_r, max_r, zcell_size))
groups, d_groups = gt.group_by_depth_range(dtime, zpressure, ndata, columns, ranges)
# ... excluding timestamps
if 'scatter' in sv:
n_std = None #to use percentile
else:
n_std = n_std
# rejecting timestamps from percentile analysis
y_avg, n_avg, n_min, n_max, n0_std, n1_std, l_arr, time_ex, \
t_nospct, z_nospct, y_nospct = cf.reject_timestamps_in_groups(groups, d_groups, n_std,
dtime, zpressure, ndata,
inpercentile)
print('{} using {} percentile of data grouped in {} dbar segments'.format(
len(zpressure) - len(z_nospct), inpercentile, zcell_size))
"""
writing timestamps to .csv file to use with data_range.py script
"""
if len(time_ex) != 0:
t_exclude = time_ex[0]
for i in range(len(time_ex))[1:len(time_ex)]:
t_exclude = '{}, {}'.format(t_exclude, time_ex[i])
stat_data = pd.DataFrame({'deployments': deployment,
'time_to_exclude': t_exclude}, index=[sv])
stat_data.to_csv(file_exclude, index=True, mode='a', header=False)
# reject time range from data portal file export
t_portal, z_portal, y_portal = cf.reject_timestamps_dataportal(subsite, r,
t_nospct, z_nospct, y_nospct)
print('{} using visual inspection of data'.format(len(z_nospct) - len(z_portal),
inpercentile, zcell_size))
# reject data in a depth range
if zdbar is not None:
y_ind = y_portal < zdbar
t_array = t_portal[y_ind]
y_array = y_portal[y_ind]
z_array = z_portal[y_ind]
else:
y_ind = []
t_array = t_portal
y_array = y_portal
z_array = z_portal
print('{} in water depth > {} dbar'.format(len(y_ind), zdbar))
"""
Plot data
"""
if len(t_array) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((sv, r))
else:
sname = '-'.join((sv, r, m))
xlabel = sv + " (" + sv_units + ")"
ylabel = y_name[0] + " (" + y_unit[0] + ")"
clabel = 'Time'
title = ' '.join((deployment, r, m))
# plot non-erroneous data
fig, ax = pf.plot_profiles(ndata, zpressure, dtime,
ylabel, xlabel, clabel, end_times, deployments, stdev=None)
ax.set_title(title, fontsize=9)
ax.plot(n_avg, y_avg, '-k')
ax.fill_betweenx(y_avg, n0_std, n1_std, color='m', alpha=0.2)
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}]'.format(
len(z) - lenfv, len(z) - lennan, len(z) - lenev, lengr, global_min, global_max) + '\n' +
('(black) data average in {} dbar segments'.format(zcell_size)) + '\n' +
('(magenta) upper and lower {} percentile envelope in {} dbar segments'.format(
inpercentile, zcell_size)),)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
fig.tight_layout()
sfile = '_'.join(('rm_erroneous_data', sname))
pf.save_fig(save_dir, sfile)
# plot excluding time ranges for suspect data
if len(z_nospct) != len(zpressure):
fig, ax = pf.plot_profiles(z_nospct, y_nospct, t_nospct,
ylabel, xlabel, clabel, end_times, deployments, stdev=None)
ax.set_title(title, fontsize=9)
leg_text = (
'removed {} in the upper and lower {} percentile of data grouped in {} dbar segments'.format(
len(zpressure) - len(z_nospct), inpercentile, zcell_size),)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
fig.tight_layout()
sfile = '_'.join(('rm_suspect_data', sname))
pf.save_fig(save_dir, sfile)
# plot excluding time ranges from data portal export
if len(z_nospct) - len(z_portal):
fig, ax = pf.plot_profiles(z_portal, y_portal, t_portal,
ylabel, xlabel, clabel, end_times, deployments, stdev=None)
ax.set_title(title, fontsize=9)
leg_text = ('excluded {} suspect data when inspected visually'.format(
len(z_nospct) - len(z_portal)),)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
fig.tight_layout()
sfile = '_'.join(('rm_v_suspect_data', sname))
pf.save_fig(save_dir, sfile)
# Plot excluding a selected depth value
if len(z_array) != len(z_array):
fig, ax = pf.plot_profiles(z_array, y_array, t_array,
ylabel, xlabel, clabel, end_times, deployments, stdev=None)
ax.set_title(title, fontsize=9)
leg_text = ('excluded {} suspect data in water depth greater than {} dbar'.format(len(y_ind), zdbar),)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
fig.tight_layout()
sfile = '_'.join(('rm_depth_range', sname))
pf.save_fig(save_dir, sfile)
def main(url_list, sDir, mDir, zcell_size, zdbar, start_time, end_time):
"""""
URL : path to instrument data by methods
sDir : path to the directory on your machine to save files
plot_type: folder name for a plot type
""" ""
rd_list = []
ms_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
if ms not in ms_list:
ms_list.append(ms)
'''
separate different instruments
'''
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
main_sensor = r.split('-')[-1]
# read in the analysis file
dr_data = cf.refdes_datareview_json(r)
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
# get science variable long names from the Data Review Database
stream_sci_vars = cd.sci_var_long_names(r)
#stream_vars = cd.var_long_names(r)
# check if the science variable long names are the same for each stream and initialize empty arrays
sci_vars_dict0 = cd.sci_var_long_names_check(stream_sci_vars)
# get the list of data files and filter out collocated instruments and other streams
datasets = []
for u in url_list:
print(u)
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
fdatasets = cf.filter_other_streams(r, ms_list, fdatasets)
# select the list of data files from the preferred dataset for each deployment
fdatasets_final = []
for ii in range(len(ps_df)):
for x in fdatasets:
if ps_df['deployment'][ii] in x and ps_df[0][ii] in x:
fdatasets_final.append(x)
# build dictionary of science data from the preferred dataset for each deployment
print('\nAppending data from files')
et = []
sci_vars_dict, y_unit, y_name = cd.append_evaluated_science_data(
sDir, ps_df, n_streams, r, fdatasets_final, sci_vars_dict0, et,
start_time, end_time)
# get end times of deployments
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = cf.get_deployment_information(
dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
"""
create a data-ranges table and figure for full data time range
"""
# create a folder to save data ranges
save_dir_stat = os.path.join(mDir, array, subsite)
cf.create_dir(save_dir_stat)
save_fdir = os.path.join(sDir, array, subsite, r, 'data_range')
cf.create_dir(save_fdir)
stat_df = pd.DataFrame()
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(vinfo['var_name'])
if len(vinfo['t']) < 1:
print('no variable data to plot')
continue
else:
sv_units = vinfo['units'][0]
fv = vinfo['fv'][0]
t = vinfo['t']
z = vinfo['values']
y = vinfo['pressure']
# Check if the array is all NaNs
if sum(np.isnan(z)) == len(z):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z[z != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
"""
clean up data
"""
# reject erroneous data
dtime, zpressure, ndata, lenfv, lennan, lenev, lengr, global_min, global_max = \
cf.reject_erroneous_data(r, sv, t, y, z, fv)
# reject timestamps from stat analysis
Dpath = '{}/{}/{}/{}/{}'.format(sDir, array, subsite, r,
'time_to_exclude')
onlyfiles = []
for item in os.listdir(Dpath):
if not item.startswith('.') and os.path.isfile(
os.path.join(Dpath, item)):
onlyfiles.append(join(Dpath, item))
dre = pd.DataFrame()
for nn in onlyfiles:
dr = pd.read_csv(nn)
dre = dre.append(dr, ignore_index=True)
drn = dre.loc[dre['Unnamed: 0'] == vinfo['var_name']]
list_time = []
for itime in drn.time_to_exclude:
ntime = itime.split(', ')
list_time.extend(ntime)
u_time_list = np.unique(list_time)
if len(u_time_list) != 0:
t_nospct, z_nospct, y_nospct = cf.reject_suspect_data(
dtime, zpressure, ndata, u_time_list)
print(
'{} using {} percentile of data grouped in {} dbar segments'
.format(
len(zpressure) - len(z_nospct), inpercentile,
zcell_size))
# reject time range from data portal file export
t_portal, z_portal, y_portal = cf.reject_timestamps_dataportal(
subsite, r, t_nospct, y_nospct, z_nospct)
print('{} using visual inspection of data'.format(
len(z_nospct) - len(z_portal), inpercentile,
zcell_size))
# reject data in a depth range
if zdbar is not None:
y_ind = y_portal < zdbar
t_array = t_portal[y_ind]
y_array = y_portal[y_ind]
z_array = z_portal[y_ind]
else:
y_ind = []
t_array = t_portal
y_array = y_portal
z_array = z_portal
print('{} in water depth > {} dbar'.format(
len(y_ind), zdbar))
if len(y_array) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((vinfo['var_name'], r))
else:
sname = '-'.join((vinfo['var_name'], r, m))
"""
create data ranges for non - pressure data only
"""
if 'pressure' in vinfo['var_name']:
pass
else:
columns = ['tsec', 'dbar', str(vinfo['var_name'])]
# create depth ranges
min_r = int(round(min(y_array) - zcell_size))
max_r = int(round(max(y_array) + zcell_size))
ranges = list(range(min_r, max_r, zcell_size))
# group data by depth
groups, d_groups = gt.group_by_depth_range(
t_array, y_array, z_array, columns, ranges)
print('writing data ranges for {}'.format(
vinfo['var_name']))
stat_data = groups.describe()[vinfo['var_name']]
stat_data.insert(loc=0,
column='parameter',
value=sv,
allow_duplicates=False)
t_deploy = deployments[0]
for i in range(
len(deployments))[1:len(deployments)]:
t_deploy = '{}, {}'.format(
t_deploy, deployments[i])
stat_data.insert(loc=1,
column='deployments',
value=t_deploy,
allow_duplicates=False)
stat_df = stat_df.append(stat_data, ignore_index=True)
"""
plot full time range free from errors and suspect data
"""
clabel = sv + " (" + sv_units + ")"
ylabel = (y_name[0][0] + " (" + y_unit[0][0] + ")")
title = ' '.join((r, m))
# plot non-erroneous -suspect data
fig, ax, bar = pf.plot_xsection(subsite,
t_array,
y_array,
z_array,
clabel,
ylabel,
inpercentile=None,
stdev=None)
ax.set_title(title, fontsize=9)
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}]'
.format(
len(z) - lenfv,
len(z) - lennan,
len(z) - lenev, lengr, global_min,
global_max) + '\n' +
('removed {} in the upper and lower {} percentile of data grouped in {} dbar segments'
.format(
len(zpressure) - len(z_nospct), inpercentile,
zcell_size)), )
ax.legend(leg_text,
loc='upper center',
bbox_to_anchor=(0.5, -0.17),
fontsize=6)
for ii in range(len(end_times)):
ax.axvline(x=end_times[ii],
color='b',
linestyle='--',
linewidth=.8)
ax.text(end_times[ii],
min(y_array) - 5,
'End' + str(deployments[ii]),
fontsize=6,
style='italic',
bbox=dict(
boxstyle='round',
ec=(0., 0.5, 0.5),
fc=(1., 1., 1.),
))
fig.tight_layout()
sfile = '_'.join(('data_range', sname))
pf.save_fig(save_fdir, sfile)
# write stat file
stat_df.to_csv('{}/{}_data_ranges.csv'.format(save_dir_stat, r),
index=True,
float_format='%11.6f')
def main(url_list, sDir, plot_type, deployment_num, start_time, end_time, preferred_only, glider, zdbar, n_std, inpercentile, zcell_size):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list and 'ENG' not in rd:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(main_sensor, fdatasets)
for fd in fdatasets_sel:
part_d = fd.split('/')[-1]
print(part_d)
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(fd)
array = subsite[0:2]
sci_vars = cf.return_science_vars(stream)
if 'CE05MOAS' in r or 'CP05MOAS' in r: # for coastal gliders, get m_water_depth for bathymetry
eng = '-'.join((r.split('-')[0], r.split('-')[1], '00-ENG000000', method, 'glider_eng'))
eng_url = [s for s in url_list if eng in s]
if len(eng_url) == 1:
eng_datasets = cf.get_nc_urls(eng_url)
# filter out collocated datasets
eng_dataset = [j for j in eng_datasets if (eng in j.split('/')[-1] and deployment in j.split('/')[-1])]
if len(eng_dataset) > 0:
ds_eng = xr.open_dataset(eng_dataset[0], mask_and_scale=False)
t_eng = ds_eng['time'].values
m_water_depth = ds_eng['m_water_depth'].values
# m_altimeter_status = 0 means a good reading (not nan or -1)
eng_ind = ds_eng['m_altimeter_status'].values == 0
m_water_depth = m_water_depth[eng_ind]
t_eng = t_eng[eng_ind]
else:
print('No engineering file for deployment {}'.format(deployment))
if deployment_num is not None:
if int(deployment.split('0')[-1]) is not deployment_num:
print(type(int(deployment.split('0')[-1])), type(deployment_num))
continue
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print('No data to plot for specified time range: ({} to {})'.format(start_time, end_time))
continue
stime = start_time.strftime('%Y-%m-%d')
etime = end_time.strftime('%Y-%m-%d')
ext = stime + 'to' + etime # .join((ds0_method, ds1_method
save_dir = os.path.join(sDir, array, subsite, refdes, plot_type, deployment, ext)
else:
save_dir = os.path.join(sDir, array, subsite, refdes, plot_type, deployment)
cf.create_dir(save_dir)
tm = ds['time'].values
# get pressure variable
ds_vars = list(ds.data_vars.keys()) + [x for x in ds.coords.keys() if 'pressure' in x]
y, y_units, press = cf.add_pressure_to_dictionary_of_sci_vars(ds)
print(y_units, press)
# press = pf.pressure_var(ds, ds_vars)
# print(press)
# y = ds[press].values
# y_units = ds[press].units
for sv in sci_vars:
print(sv)
if 'sci_water_pressure' not in sv:
z = ds[sv].values
fv = ds[sv]._FillValue
z_units = ds[sv].units
# Check if the array is all NaNs
if sum(np.isnan(z)) == len(z):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z[z != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
"""
clean up data
"""
# reject erroneous data
dtime, zpressure, ndata, lenfv, lennan, lenev, lengr, global_min, global_max = \
cf.reject_erroneous_data(r, sv, tm, y, z, fv)
# get rid of 0.0 data
if 'CTD' in r:
ind = zpressure > 0.0
else:
ind = ndata > 0.0
lenzero = np.sum(~ind)
dtime = dtime[ind]
zpressure = zpressure[ind]
ndata = ndata[ind]
# creating data groups
columns = ['tsec', 'dbar', str(sv)]
min_r = int(round(min(zpressure) - zcell_size))
max_r = int(round(max(zpressure) + zcell_size))
ranges = list(range(min_r, max_r, zcell_size))
groups, d_groups = gt.group_by_depth_range(dtime, zpressure, ndata, columns, ranges)
# rejecting timestamps from percentile analysis
y_avg, n_avg, n_min, n_max, n0_std, n1_std, l_arr, time_ex = cf.reject_timestamps_in_groups(
groups, d_groups, n_std, inpercentile)
t_nospct, z_nospct, y_nospct = cf.reject_suspect_data(dtime, zpressure, ndata, time_ex)
print('removed {} data points using {} percentile of data grouped in {} dbar segments'.format(
len(zpressure) - len(z_nospct), inpercentile, zcell_size))
# reject time range from data portal file export
t_portal, z_portal, y_portal = cf.reject_timestamps_dataportal(subsite, r,
t_nospct, y_nospct, z_nospct)
print('removed {} data points using visual inspection of data'.format(len(z_nospct) - len(z_portal)))
# reject data in a depth range
if zdbar:
y_ind = y_portal < zdbar
n_zdbar = np.sum(~y_ind)
t_array = t_portal[y_ind]
y_array = y_portal[y_ind]
z_array = z_portal[y_ind]
else:
n_zdbar = 0
t_array = t_portal
y_array = y_portal
z_array = z_portal
print('{} in water depth > {} dbar'.format(n_zdbar, zdbar))
"""
Plot data
"""
if len(dtime) > 0:
sname = '-'.join((r, method, sv))
clabel = sv + " (" + z_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
if glider == 'no':
t_eng = None
m_water_depth = None
# plot non-erroneous data
fig, ax, bar = pf.plot_xsection(subsite, dtime, zpressure, ndata, clabel, ylabel,
t_eng, m_water_depth, inpercentile, stdev=None)
t0 = pd.to_datetime(dtime.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(dtime.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method)) + '\n' + t0 + ' to ' + t1
ax.set_title(title, fontsize=9)
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} zeros'.format(lenfv, lennan, lenev, lengr, global_min, global_max, lenzero),
)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
fig.tight_layout()
sfile = '_'.join(('rm_erroneous_data', sname))
pf.save_fig(save_dir, sfile)
# plots removing all suspect data
if len(t_array) > 0:
if len(t_array) != len(dtime):
# plot bathymetry only within data time ranges
if glider == 'yes':
eng_ind = (t_eng >= np.min(t_array)) & (t_eng <= np.max(t_array))
t_eng = t_eng[eng_ind]
m_water_depth = m_water_depth[eng_ind]
fig, ax, bar = pf.plot_xsection(subsite, t_array, y_array, z_array, clabel, ylabel,
t_eng, m_water_depth, inpercentile, stdev=None)
t0 = pd.to_datetime(t_array.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(t_array.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method)) + '\n' + t0 + ' to ' + t1
ax.set_title(title, fontsize=9)
if zdbar:
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} zeros'.format(lenfv, lennan, lenev, lengr, global_min, global_max, lenzero)
+ '\nremoved {} in the upper and lower {}th percentile of data grouped in {} dbar segments'.format(
len(zpressure) - len(z_nospct), inpercentile, zcell_size)
+ '\nexcluded {} suspect data points when inspected visually'.format(
len(z_nospct) - len(z_portal))
+ '\nexcluded {} suspect data in water depth greater than {} dbar'.format(n_zdbar,
zdbar),
)
else:
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} zeros'.format(lenfv, lennan, lenev, lengr, global_min, global_max, lenzero)
+ '\nremoved {} in the upper and lower {}th percentile of data grouped in {} dbar segments'.format(
len(zpressure) - len(z_nospct), inpercentile, zcell_size)
+ '\nexcluded {} suspect data points when inspected visually'.format(
len(z_nospct) - len(z_portal)),
)
ax.legend(leg_text, loc='upper center', bbox_to_anchor=(0.5, -0.17), fontsize=6)
fig.tight_layout()
sfile = '_'.join(('rm_suspect_data', sname))
pf.save_fig(save_dir, sfile)
def main(sDir, url_list, start_time, end_time, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
fdatasets = np.unique(fdatasets).tolist()
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, fdatasets)
for fd in fdatasets:
if '_blank' not in fd:
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
ds_vars = list(ds.data_vars.keys()) + [x for x in ds.coords.keys() if 'pressure' in x] # get pressure variable from coordinates
#raw_vars = cf.return_raw_vars(ds_vars)
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print('No data to plot for specified time range: ({} to {})'.format(start_time, end_time))
continue
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(fd)
if 'NUTNR' in refdes:
vars = cf.return_science_vars(stream)
else:
vars = cf.return_raw_vars(ds_vars)
print('\nPlotting {} {}'.format(r, deployment))
array = subsite[0:2]
filename = '_'.join(fname.split('_')[:-1])
save_dir = os.path.join(sDir, array, subsite, refdes, 'timeseries_plots', deployment)
cf.create_dir(save_dir)
tm = ds['time'].values
t0 = pd.to_datetime(tm.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(tm.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method))
for var in vars:
print(var)
if var != 'id':
y = ds[var]
try:
fv = y._FillValue
except AttributeError:
fv = np.nan
if len(y.dims) == 1:
# Check if the array is all NaNs
if sum(np.isnan(y.values)) == len(y.values):
print('Array of all NaNs - skipping plot.')
# Check if the array is all fill values
elif len(y[y != fv]) == 0:
print('Array of all fill values - skipping plot.')
else:
# reject fill values
ind = y.values != fv
t = tm[ind]
y = y[ind]
# Plot all data
fig, ax = pf.plot_timeseries(t, y, y.name, stdev=None)
ax.set_title((title + '\n' + t0 + ' - ' + t1), fontsize=9)
sfile = '-'.join((filename, y.name, t0[:10]))
pf.save_fig(save_dir, sfile)
# Plot data with outliers removed
fig, ax = pf.plot_timeseries(t, y, y.name, stdev=5)
ax.set_title((title + '\n' + t0 + ' - ' + t1), fontsize=9)
sfile = '-'.join((filename, y.name, t0[:10])) + '_rmoutliers'
pf.save_fig(save_dir, sfile)
def main(url_list, sDir, plot_type, start_time, end_time, deployment_num):
for i, u in enumerate(url_list):
elements = u.split('/')[-2].split('-')
r = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = u.split(r + '-')[1].split('/')[0]
subsite = r.split('-')[0]
array = subsite[0:2]
main_sensor = r.split('-')[-1]
datasets = cf.get_nc_urls([u])
datasets_sel = cf.filter_collocated_instruments(main_sensor, datasets)
save_dir = os.path.join(sDir, array, subsite, r, plot_type)
cf.create_dir(save_dir)
sname = '-'.join((r, ms, 'track'))
print('Appending....')
sh = pd.DataFrame()
deployments = []
end_times = []
for ii, d in enumerate(datasets_sel):
print('\nDataset {} of {}: {}'.format(ii + 1, len(datasets_sel),
d.split('/')[-1]))
ds = xr.open_dataset(d, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'.
format(start_time, end_time))
continue
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(
d)
if deployment_num is not None:
if int(deployment.split('0')[-1]) is not deployment_num:
print(type(int(deployment.split('0')[-1])),
type(deployment_num))
continue
# get end times of deployments
ps_df, n_streams = cf.get_preferred_stream_info(r)
dr_data = cf.refdes_datareview_json(r)
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = cf.get_deployment_information(
dr_data, int(deploy[-4:]))
if int(deploy[-4:]) not in deployments:
deployments.append(int(deploy[-4:]))
if pd.to_datetime(deploy_info['stop_date']) not in end_times:
end_times.append(pd.to_datetime(deploy_info['stop_date']))
data = {'lat': ds['lat'].values, 'lon': ds['lon'].values}
new_r = pd.DataFrame(data,
columns=['lat', 'lon'],
index=ds['time'].values)
sh = sh.append(new_r)
xD = sh.lon.values
yD = sh.lat.values
tD = sh.index.values
clabel = 'Time'
ylabel = 'Latitude'
xlabel = 'Longitude'
fig, ax = pf.plot_profiles(xD,
yD,
tD,
ylabel,
xlabel,
clabel,
end_times,
deployments,
stdev=None)
ax.invert_yaxis()
ax.set_title('Glider Track - ' + r + '\n' + 'x: platform location',
fontsize=9)
ax.set_xlim(-71.75, -69.75)
ax.set_ylim(38.75, 40.75)
#cbar.ax.set_yticklabels(end_times)
# add Pioneer glider sampling area
ax.add_patch(
Rectangle((-71.5, 39.0),
1.58,
1.67,
linewidth=3,
edgecolor='b',
facecolor='none'))
ax.text(-71,
40.6,
'Pioneer Glider Sampling Area',
color='blue',
fontsize=8)
# add Pioneer AUV sampling area
# ax.add_patch(Rectangle((-71.17, 39.67), 0.92, 1.0, linewidth=3, edgecolor='m', facecolor='none'))
array_loc = cf.return_array_subsites_standard_loc(array)
ax.scatter(array_loc.lon,
array_loc.lat,
s=40,
marker='x',
color='k',
alpha=0.3)
#ax.legend(legn, array_loc.index, scatterpoints=1, loc='lower left', ncol=4, fontsize=8)
pf.save_fig(save_dir, sname)
def main(sDir, url_list, start_time, end_time, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list and 'OPTAA' in rd:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
for ud in udatasets: # filter out collocated data files
if 'OPTAA' in ud.split('/')[-1]:
datasets.append(ud)
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
fdatasets = np.unique(fdatasets).tolist()
for fd in fdatasets:
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print('No data to plot for specified time range: ({} to {})'.format(start_time, end_time))
continue
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(fd)
#sci_vars = cf.return_science_vars(stream)
sci_vars = ['optical_absorption', 'beam_attenuation']
print('\nPlotting {} {}'.format(r, deployment))
array = subsite[0:2]
filename = '_'.join(fname.split('_')[:-1])
save_dir = os.path.join(sDir, array, subsite, refdes, 'timeseries_plots')
cf.create_dir(save_dir)
tm = ds['time'].values
t0 = pd.to_datetime(tm.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(tm.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method))
# # add chl-a data from the collocated fluorometer
# flor_url = [s for s in url_list if r.split('-')[0] in s and 'FLOR' in s]
# if len(flor_url) == 1:
# flor_datasets = cf.get_nc_urls(flor_url)
# # filter out collocated datasets
# flor_dataset = [j for j in flor_datasets if ('FLOR' in j.split('/')[-1] and deployment in j.split('/')[-1])]
# if len(flor_dataset) > 0:
# ds_flor = xr.open_dataset(flor_dataset[0], mask_and_scale=False)
# ds_flor = ds_flor.swap_dims({'obs': 'time'})
# flor_t0 = dt.datetime.strptime(t0, '%Y-%m-%dT%H:%M:%S')
# flor_t1 = dt.datetime.strptime(t1, '%Y-%m-%dT%H:%M:%S')
# ds_flor = ds_flor.sel(time=slice(flor_t0, flor_t1))
# t_flor = ds_flor['time'].values
# flor_sci_vars = cf.return_science_vars(ds_flor.stream)
# for fsv in flor_sci_vars:
# if ds_flor[fsv].long_name == 'Chlorophyll-a Concentration':
# chla = ds_flor[fsv]
for var in sci_vars:
print(var)
if var == 'optical_absorption':
wv = ds['wavelength_a'].values
else:
wv = ds['wavelength_c'].values
vv = ds[var]
fv = vv._FillValue
fig1, ax1 = plt.subplots()
fig2, ax2 = plt.subplots()
plotting = [] # keep track if anything is plotted
wavelengths = []
iwavelengths = []
for i in range(len(wv)):
if (wv[i] > 671.) and (wv[i] < 679.):
wavelengths.append(wv[i])
iwavelengths.append(i)
colors = ['purple', 'green', 'orange']
for iw in range(len(iwavelengths)):
v = vv.sel(wavelength=iwavelengths[iw]).values
n_all = len(v)
n_nan = np.sum(np.isnan(v))
# convert fill values to nans
v[v == fv] = np.nan
n_fv = np.sum(np.isnan(v)) - n_nan
if n_nan + n_fv < n_all:
# plot before global ranges are removed
#fig, ax = pf.plot_optaa(tm, v, vv.name, vv.units)
plotting.append('yes')
ax1.scatter(tm, v, c=colors[iw], label='{} nm'.format(wavelengths[iw]),
marker='.', s=1)
# reject data outside of global ranges
[g_min, g_max] = cf.get_global_ranges(r, var)
if g_min is not None and g_max is not None:
v[v < g_min] = np.nan
v[v > g_max] = np.nan
n_grange = np.sum(np.isnan(v)) - n_fv - n_nan
else:
n_grange = 'no global ranges'
# plot after global ranges are removed
ax2.scatter(tm, v, c=colors[iw], label='{} nm: rm {} GR'.format(wavelengths[iw], n_grange),
marker='.', s=1)
# if iw == len(wavelengths) - 1:
# ax2a = ax2.twinx()
# ax2a.scatter(t_flor, chla.values, c='lime', marker='.', s=1)
# ax2a.set_ylabel('Fluorometric Chl-a ({})'.format(chla.units))
if len(plotting) > 0:
ax1.grid()
pf.format_date_axis(ax1, fig1)
ax1.legend(loc='best', fontsize=7)
ax1.set_ylabel((var + " (" + vv.units + ")"), fontsize=9)
ax1.set_title((title + '\n' + t0 + ' - ' + t1), fontsize=9)
sfile = '-'.join((filename, var, t0[:10]))
save_file = os.path.join(save_dir, sfile)
fig1.savefig(str(save_file), dpi=150)
ax2.grid()
pf.format_date_axis(ax2, fig2)
ax2.legend(loc='best', fontsize=7)
ax2.set_ylabel((var + " (" + vv.units + ")"), fontsize=9)
title_gr = 'GR: global ranges'
ax2.set_title((title + '\n' + t0 + ' - ' + t1 + '\n' + title_gr), fontsize=9)
sfile2 = '-'.join((filename, var, t0[:10], 'rmgr'))
save_file2 = os.path.join(save_dir, sfile2)
fig2.savefig(str(save_file2), dpi=150)
plt.close('all')
def main(sDir, url_list, start_time, end_time):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(main_sensor, fdatasets)
# get science variable long names from the Data Review Database
#stream_sci_vars = cd.sci_var_long_names(r)
if 'SPKIR' in r or 'PRESF' in r: # only get the main science variable for SPKIR
stream_vars = cd.sci_var_long_names(r)
else:
stream_vars = var_long_names(r)
# check if the science variable long names are the same for each stream and initialize empty arrays
sci_vars_dict = cd.sci_var_long_names_check(stream_vars)
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
# build dictionary of science data from the preferred dataset for each deployment
print('\nAppending data from files')
et = []
sci_vars_dict, __, __ = cd.append_science_data(ps_df, n_streams, r, fdatasets_sel, sci_vars_dict, et, start_time, end_time)
# get end times of deployments
dr_data = cf.refdes_datareview_json(r)
deployments = []
dend_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
dend_times.append(pd.to_datetime(deploy_info['stop_date']))
subsite = r.split('-')[0]
array = subsite[0:2]
save_dir = os.path.join(sDir, array, subsite, r, 'timeseries_plots_preferred_all')
cf.create_dir(save_dir)
print('\nPlotting data')
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(sv)
if 'SPKIR' in r:
fv_lst = np.unique(vinfo['fv']).tolist()
if len(fv_lst) == 1:
fill_value = fv_lst[0]
else:
print(fv_lst)
print('No unique fill value for {}'.format(sv))
sv_units = np.unique(vinfo['units']).tolist()
t = vinfo['t']
if len(t) > 1:
data = vinfo['values']
[dd_data, g_min, g_max] = index_dataset_2d(r, 'spkir_abj_cspp_downwelling_vector', data, fill_value)
t0 = pd.to_datetime(min(t)).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(t)).strftime('%Y-%m-%dT%H:%M:%S')
deploy_final = vinfo['deployments']
deploy = list(np.unique(deploy_final))
deployments = [int(dd) for dd in deploy]
sname = '-'.join((r, sv))
fig, ax = pf.plot_spkir(t, dd_data, sv, sv_units[0])
ax.set_title((r + '\nDeployments: ' + str(sorted(deployments)) + '\n' + t0 + ' - ' + t1 + '\n'
+ 'removed global ranges +/- [{} - {}]'.format(g_min, g_max)), fontsize=8)
for etimes in dend_times:
ax.axvline(x=etimes, color='k', linestyle='--', linewidth=.6)
pf.save_fig(save_dir, sname)
# plot each wavelength
wavelengths = ['412nm', '443nm', '490nm', '510nm', '555nm', '620nm', '683nm']
for wvi in range(len(dd_data)):
fig, ax = pf.plot_spkir_wv(t, dd_data[wvi], sv, sv_units[0], wvi)
ax.set_title(
(r + '\nDeployments: ' + str(sorted(deployments)) + '\n' + t0 + ' - ' + t1 + '\n'
+ 'removed global ranges +/- [{} - {}]'.format(g_min, g_max)), fontsize=8)
for etimes in dend_times:
ax.axvline(x=etimes, color='k', linestyle='--', linewidth=.6)
snamewvi = '-'.join((sname, wavelengths[wvi]))
pf.save_fig(save_dir, snamewvi)
elif 'presf_abc_wave_burst' in m:
fv_lst = np.unique(vinfo['fv']).tolist()
if len(fv_lst) == 1:
fill_value = fv_lst[0]
else:
print(fv_lst)
print('No unique fill value for {}'.format(sv))
sv_units = np.unique(vinfo['units']).tolist()
t = vinfo['t']
if len(t) > 1:
data = vinfo['values']
[dd_data, g_min, g_max] = index_dataset_2d(r, 'presf_wave_burst_pressure', data, fill_value)
t0 = pd.to_datetime(min(t)).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(t)).strftime('%Y-%m-%dT%H:%M:%S')
deploy_final = vinfo['deployments']
deploy = list(np.unique(deploy_final))
deployments = [int(dd) for dd in deploy]
sname = '-'.join((r, sv))
fig, ax = pf.plot_presf_2d(t, dd_data, sv, sv_units[0])
ax.set_title((r + '\nDeployments: ' + str(sorted(deployments)) + '\n' + t0 + ' - ' + t1 + '\n'
+ 'removed global ranges +/- [{} - {}]'.format(g_min, g_max)), fontsize=8)
for etimes in dend_times:
ax.axvline(x=etimes, color='k', linestyle='--', linewidth=.6)
pf.save_fig(save_dir, sname)
else:
if type(vinfo['values']) != dict: # if the variable is not a 2D array
if 'Spectra' not in sv:
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
sv_name = vinfo['var_name']
t0 = pd.to_datetime(min(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
x = vinfo['t']
y = vinfo['values']
# reject NaNs and values of 0.0
nan_ind = (~np.isnan(y)) & (y != 0.0)
x_nonan = x[nan_ind]
y_nonan = y[nan_ind]
# reject fill values
fv_ind = y_nonan != vinfo['fv'][0]
x_nonan_nofv = x_nonan[fv_ind]
y_nonan_nofv = y_nonan[fv_ind]
# reject extreme values
Ev_ind = cf.reject_extreme_values(y_nonan_nofv)
y_nonan_nofv_nE = y_nonan_nofv[Ev_ind]
x_nonan_nofv_nE = x_nonan_nofv[Ev_ind]
# reject values outside global ranges:
global_min, global_max = cf.get_global_ranges(r, sv_name)
if any(e is None for e in [global_min, global_max]):
y_nonan_nofv_nE_nogr = y_nonan_nofv_nE
x_nonan_nofv_nE_nogr = x_nonan_nofv_nE
else:
gr_ind = cf.reject_global_ranges(y_nonan_nofv_nE, global_min, global_max)
y_nonan_nofv_nE_nogr = y_nonan_nofv_nE[gr_ind]
x_nonan_nofv_nE_nogr = x_nonan_nofv_nE[gr_ind]
if len(y_nonan_nofv) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((r, sv))
else:
sname = '-'.join((r, m, sv))
plt_deploy = [int(x) for x in list(np.unique(vinfo['deployments']))]
# plot hourly averages for cabled and FDCHP data
if 'streamed' in sci_vars_dict[list(sci_vars_dict.keys())[0]]['ms'][0] or 'FDCHP' in r:
sname = '-'.join((sname, 'hourlyavg'))
df = pd.DataFrame({'dfx': x_nonan_nofv_nE_nogr, 'dfy': y_nonan_nofv_nE_nogr})
dfr = df.resample('H', on='dfx').mean()
# Plot all data
fig, ax = pf.plot_timeseries_all(dfr.index, dfr['dfy'], sv, sv_units, stdev=None)
ax.set_title((r + '\nDeployments: ' + str(plt_deploy) + '\n' + t0 + ' - ' + t1),
fontsize=8)
# if plotting a specific time range, plot deployment lines only for those deployments
if type(start_time) == dt.datetime:
for e in list(np.unique(vinfo['deployments'])):
etime = dend_times[int(e) - 1]
ax.axvline(x=etime, color='b', linestyle='--', linewidth=.6)
else:
for etime in dend_times:
ax.axvline(x=etime, color='b', linestyle='--', linewidth=.6)
pf.save_fig(save_dir, sname)
else:
# Plot all data
fig, ax = pf.plot_timeseries_all(x_nonan_nofv, y_nonan_nofv, sv, sv_units, stdev=None)
ax.set_title((r + '\nDeployments: ' + str(plt_deploy) + '\n' + t0 + ' - ' + t1),
fontsize=8)
# if plotting a specific time range, plot deployment lines only for those deployments
if type(start_time) == dt.datetime:
# for e in list(np.unique(vinfo['deployments'])):
# etime = dend_times[int(e) - 1]
# ax.axvline(x=etime, color='b', linestyle='--', linewidth=.6)
etime = dend_times[int(list(np.unique(vinfo['deployments']))[0]) - 1]
ax.axvline(x=etime, color='b', linestyle='--', linewidth=.6)
else:
for etime in dend_times:
ax.axvline(x=etime, color='b', linestyle='--', linewidth=.6)
# if not any(e is None for e in [global_min, global_max]):
# ax.axhline(y=global_min, color='r', linestyle='--', linewidth=.6)
# ax.axhline(y=global_max, color='r', linestyle='--', linewidth=.6)
# else:
# maxpoint = x[np.argmax(y_nonan_nofv)], max(y_nonan_nofv)
# ax.annotate('No Global Ranges', size=8,
# xy=maxpoint, xytext=(5, 5), textcoords='offset points')
pf.save_fig(save_dir, sname)
# Plot data with outliers removed
fig, ax = pf.plot_timeseries_all(x_nonan_nofv_nE_nogr, y_nonan_nofv_nE_nogr, sv, sv_units,
stdev=5)
ax.set_title((r + '\nDeployments: ' + str(plt_deploy) + '\n' + t0 + ' - ' + t1),
fontsize=8)
# if plotting a specific time range, plot deployment lines only for those deployments
if type(start_time) == dt.datetime:
# for e in list(np.unique(vinfo['deployments'])):
# etime = dend_times[int(e) - 1]
# ax.axvline(x=etime, color='b', linestyle='--', linewidth=.6)
etime = dend_times[int(list(np.unique(vinfo['deployments']))[0]) - 1]
ax.axvline(x=etime, color='b', linestyle='--', linewidth=.6)
else:
for etime in dend_times:
ax.axvline(x=etime, color='b', linestyle='--', linewidth=.6)
# if not any(e is None for e in [global_min, global_max]):
# ax.axhline(y=global_min, color='r', linestyle='--', linewidth=.6)
# ax.axhline(y=global_max, color='r', linestyle='--', linewidth=.6)
# else:
# maxpoint = x[np.argmax(y_nonan_nofv_nE_nogr)], max(y_nonan_nofv_nE_nogr)
# ax.annotate('No Global Ranges', size=8,
# xy=maxpoint, xytext=(5, 5), textcoords='offset points')
sfile = '_'.join((sname, 'rmoutliers'))
pf.save_fig(save_dir, sfile)
def main(sDir, url_list, start_time, end_time, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
fdatasets = np.unique(fdatasets).tolist()
for fd in fdatasets:
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'.
format(start_time, end_time))
continue
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(
fd)
sci_vars = cf.return_science_vars(stream)
print('\nPlotting {} {}'.format(r, deployment))
array = subsite[0:2]
filename = '_'.join(fname.split('_')[:-1])
save_dir = os.path.join(sDir, array, subsite, refdes,
'timeseries_plots')
cf.create_dir(save_dir)
tm = ds['time'].values
t0 = pd.to_datetime(tm.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(tm.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method))
# -------- plot entire deployment --------
for var in sci_vars:
print(var)
vv = ds[var]
fv = vv._FillValue
# need to round SPKIR values to 1 decimal place to match the global ranges. otherwise, values that
# round to zero (e.g. 1.55294e-05) will be excluded by the global range test
# v = np.round(vv.values.T, 1) # .T = transpose 2D array
v = vv.values.T
n_nan = np.sum(np.isnan(v))
# convert fill values to nans
v[v == fv] = np.nan
n_fv = np.sum(np.isnan(v)) - n_nan
# plot before global ranges are removed
fig, ax = pf.plot_spkir(tm, v, vv.name, vv.units)
ax.set_title((title + '\n' + t0 + ' - ' + t1), fontsize=9)
sfile = '-'.join((filename, var, t0[:10]))
pf.save_fig(save_dir, sfile)
# reject data outside of global ranges
[g_min, g_max] = cf.get_global_ranges(r, var)
if g_min is not None and g_max is not None:
v[v < g_min] = np.nan
v[v > g_max] = np.nan
n_grange = np.sum(np.isnan(v)) - n_fv - n_nan
else:
n_grange = 'no global ranges'
# plot after global ranges are removed
fig, ax = pf.plot_spkir(tm, v, vv.name, vv.units)
title2 = 'removed: {} global ranges [{}, {}]'.format(
n_grange, g_min, g_max)
ax.set_title((title + '\n' + t0 + ' - ' + t1 + '\n' + title2),
fontsize=9)
sfile = '-'.join((filename, var, t0[:10], 'rmgr'))
pf.save_fig(save_dir, sfile)
# -------- break the deployment into months and plot --------
save_dir = os.path.join(sDir, array, subsite, refdes,
'timeseries_plots', 'monthly')
cf.create_dir(save_dir)
# create list of start and end dates
dt_start = dt.datetime.strptime(t0, '%Y-%m-%dT%H:%M:%S')
dt_end = dt.datetime.strptime(t1, '%Y-%m-%dT%H:%M:%S')
start_dates = [dt_start.strftime('%m-%d-%YT00:00:00')]
end_dates = []
ts1 = dt_start
while ts1 <= dt_end:
ts2 = ts1 + dt.timedelta(days=1)
if ts2.month != ts1.month:
start_dates.append(ts2.strftime('%m-%d-%YT00:00:00'))
end_dates.append(ts1.strftime('%m-%d-%YT23:59:59'))
ts1 = ts2
end_dates.append(dt_end.strftime('%m-%d-%YT23:59:59'))
for sd, ed in zip(start_dates, end_dates):
sd_format = dt.datetime.strptime(sd, '%m-%d-%YT%H:%M:%S')
ed_format = dt.datetime.strptime(ed, '%m-%d-%YT%H:%M:%S')
ds_month = ds.sel(time=slice(sd_format, ed_format))
if len(ds_month['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'.
format(sd, ed))
continue
tm = ds_month['time'].values
t0 = pd.to_datetime(tm.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(tm.max()).strftime('%Y-%m-%dT%H:%M:%S')
for var in sci_vars:
print(var)
vv = ds_month[var]
fv = vv._FillValue
v = vv.values.T # transpose 2D array
n_nan = np.sum(np.isnan(v))
# convert fill values to nans
v[v == fv] = np.nan
n_fv = np.sum(np.isnan(v)) - n_nan
# reject data outside of global ranges
[g_min, g_max] = cf.get_global_ranges(r, var)
if g_min is not None and g_max is not None:
v[v < g_min] = np.nan
v[v > g_max] = np.nan
n_grange = np.sum(np.isnan(v)) - n_fv - n_nan
else:
n_grange = 'no global ranges'
# plot after global ranges are removed
fig, ax = pf.plot_spkir(tm, v, vv.name, vv.units)
title2 = 'removed: {} global ranges [{}, {}]'.format(
n_grange, g_min, g_max)
ax.set_title(
(title + '\n' + t0 + ' - ' + t1 + '\n' + title2),
fontsize=9)
sfile = '-'.join((filename, var, t0[:7], 'rmgr'))
pf.save_fig(save_dir, sfile)
def main(url_list, sDir, plot_type, start_time, end_time, deployment_num, bfiles):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
if 'ENG' not in r:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
if 'bottom_track_earth' not in splitter[-1]:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join((spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(main_sensor, fdatasets)
subsite = r.split('-')[0]
array = subsite[0:2]
save_dir = os.path.join(sDir, array, subsite, r, plot_type)
cf.create_dir(save_dir)
sname = '_'.join((r, plot_type))
sh = pd.DataFrame()
deployments = []
for ii, d in enumerate(fdatasets_sel):
print('\nDataset {} of {}: {}'.format(ii + 1, len(fdatasets_sel), d.split('/')[-1]))
deploy = d.split('/')[-1].split('_')[0]
if deployment_num:
if int(deploy[-4:]) is not deployment_num:
continue
ds = xr.open_dataset(d, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print('No data to plot for specified time range: ({} to {})'.format(start_time, end_time))
continue
try:
ds_lat = ds['lat'].values
except KeyError:
ds_lat = None
print('No latitude variable in file')
try:
ds_lon = ds['lon'].values
except KeyError:
ds_lon = None
print('No longitude variable in file')
if ds_lat is not None and ds_lon is not None:
data = {'lat': ds_lat, 'lon': ds_lon}
new_r = pd.DataFrame(data, columns=['lat', 'lon'], index=ds['time'].values)
sh = sh.append(new_r)
# append the deployments that are actually plotted
if int(deploy[-4:]) not in deployments:
deployments.append(int(deploy[-4:]))
# plot data by deployment
sfile = '-'.join((deploy, sname))
if array == 'CE':
ttl = 'Glider Track - ' + r + ' - ' + deploy + '\nx: Mooring Locations'
else:
ttl = 'Glider Track - ' + r + ' - ' + deploy + '\nx: Mooring Locations' + '\n blue box: Glider Sampling Area'
#fig, ax = pf.plot_profiles(ds_lon, ds_lat, ds['time'].values, ylabel, xlabel, clabel, stdev=None)
plot_map(save_dir, sfile, ttl, ds_lon, ds_lat, ds['time'].values, array, bfiles, plot_type)
sh = sh.resample('H').median() # resample hourly
xD = sh.lon.values
yD = sh.lat.values
tD = sh.index.values
title = 'Glider Track - ' + r + '\nDeployments: ' + str(deployments) + ' x: Mooring Locations' + '\n blue box: Glider Sampling Area'
save_dir_main = os.path.join(sDir, array, subsite, r)
plot_map(save_dir_main, sname, title, xD, yD, tD, array, bfiles, plot_type, add_box='yes')
def main(sDir, url_list, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
datasets = []
for u in url_list:
print(u)
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
if preferred_only == 'yes':
ps_df, n_streams = cf.get_preferred_stream_info(r)
fdatasets = []
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, fdatasets)
save_dir = os.path.join(sDir, array, subsite, r,
'preferred_method_plots')
cf.create_dir(save_dir)
# get the preferred stream information
fig, ax = pyplot.subplots(nrows=len(fdatasets), ncols=1, sharey=True)
fig.tight_layout(pad=0.4, w_pad=0.5, h_pad=1.0)
fig0, ax0 = pyplot.subplots(nrows=len(fdatasets), ncols=1, sharey=True)
fig0.tight_layout()
fig1, ax1 = pyplot.subplots(nrows=len(fdatasets), ncols=1, sharey=True)
fig1.tight_layout()
fig2, ax2 = pyplot.subplots(nrows=len(fdatasets), ncols=1, sharey=True)
fig2.tight_layout()
fig3, ax3 = pyplot.subplots(nrows=len(fdatasets), ncols=1, sharey=True)
fig3.tight_layout()
fig4, ax4 = pyplot.subplots(nrows=len(fdatasets), ncols=1, sharey=True)
fig4.tight_layout()
for ii in range(len(fdatasets)):
print('\n', fdatasets[ii])
deployment = fdatasets[ii].split('/')[-1].split('_')[0].split(
'deployment')[-1]
deployment = int(deployment)
ds = xr.open_dataset(fdatasets[ii], mask_and_scale=False)
time = ds['time'].values
sci_var = cf.return_science_vars(ds.stream)
# Plot pressure
z_name = [z_var for z_var in sci_var if 'pressure' in z_var]
z = ds[z_name[0]].values
z_unit = ds[z_name[0]].units
ax1[ii].plot(time,
z,
'b-',
linestyle='--',
linewidth=.6,
label='V')
ax1[ii].set_ylabel(str(deployment),
rotation=0,
fontsize=8,
color='b',
labelpad=11)
ax1[ii].yaxis.set_label_position("right")
ax1[ii].tick_params(which='both',
color='r',
labelsize=7,
labelcolor='m',
pad=0.1,
length=1,
rotation=0)
if ii < len(fdatasets) - 1:
ax1[ii].set_xlabel(' ')
else:
ax1[ii].set_xlabel('Time', rotation=0, fontsize=8, color='b')
if ii == 0:
ax1[ii].set_title(r + ' - Pressure ' + z_unit, fontsize=8)
sfile = 'pressure_plots'
save_file = os.path.join(save_dir, sfile)
fig1.savefig(str(save_file), dpi=150)
# non science veriable
# According to VELPT manufacturer, data are suspect when this instrument is tilted more than 20 degrees
# redmine ticket: Marine Hardware #12960
roll = ds['roll_decidegree'].values
roll_unit = ds['roll_decidegree'].units
pitch = ds['pitch_decidegree'].values
pitch_units = ds['pitch_decidegree'].units
headng = ds['heading_decidegree'].values
headng_units = ds['heading_decidegree'].values
tilt_ind = np.logical_or(pitch > 200, roll > 200)
pitch_fit = pitch[tilt_ind]
roll_fit = roll[tilt_ind]
# plot roll
ax2[ii].plot(time,
roll,
'b-',
linestyle='--',
linewidth=.6,
label='Roll')
ax2[ii].plot(time[tilt_ind],
roll_fit,
'g.',
linestyle='None',
marker='.',
markersize=0.5,
label='Roll < 200')
ax2[ii].set_ylabel(str(deployment),
rotation=0,
fontsize=8,
color='b',
labelpad=11)
ax2[ii].yaxis.set_label_position("right")
ax2[ii].tick_params(which='both',
color='r',
labelsize=7,
labelcolor='m',
pad=0.1,
length=1,
rotation=0)
if ii < len(fdatasets) - 1:
ax2[ii].set_xlabel(' ')
else:
ax2[ii].set_xlabel('Time', rotation=0, fontsize=8, color='b')
if ii == 0:
ax2[ii].set_title(r + ' - Roll ' + roll_unit, fontsize=8)
leg2 = ax2[ii].legend(fontsize=6,
bbox_to_anchor=(0., 0.80, 1., .102),
loc=3,
ncol=3,
mode="expand",
borderaxespad=0.)
leg2._drawFrame = False
sfile = 'roll_plots'
save_file = os.path.join(save_dir, sfile)
fig2.savefig(str(save_file), dpi=150)
# plot pitch
ax3[ii].plot(time,
pitch,
'b-',
linestyle='--',
linewidth=.6,
label='Roll')
ax3[ii].plot(time[tilt_ind],
pitch_fit,
'g.',
linestyle='None',
marker='.',
markersize=0.5,
label='Roll < 200')
ax3[ii].set_ylabel(str(deployment),
rotation=0,
fontsize=8,
color='b',
labelpad=11)
ax3[ii].yaxis.set_label_position("right")
ax3[ii].tick_params(which='both',
color='r',
labelsize=7,
labelcolor='m',
pad=0.1,
length=1,
rotation=0)
if ii < len(fdatasets) - 1:
ax3[ii].set_xlabel(' ')
else:
ax3[ii].set_xlabel('Time', rotation=0, fontsize=8, color='b')
if ii == 0:
ax3[ii].set_title(r + ' - Pitch ' + roll_unit, fontsize=8)
leg3 = ax2[ii].legend(fontsize=6,
bbox_to_anchor=(0., 0.80, 1., .102),
loc=3,
ncol=3,
mode="expand",
borderaxespad=0.)
leg3._drawFrame = False
sfile = 'pitch_plots'
save_file = os.path.join(save_dir, sfile)
fig3.savefig(str(save_file), dpi=150)
# plot heading
ax4[ii].plot(time,
headng,
'b-',
linestyle='None',
marker='.',
markersize=0.5,
label='Roll')
ax4[ii].plot(time[tilt_ind],
headng[tilt_ind],
'g.',
linestyle='None',
marker='.',
markersize=0.5,
label='Roll < 200')
ax4[ii].set_ylabel(str(deployment),
rotation=0,
fontsize=8,
color='b',
labelpad=11)
ax4[ii].yaxis.set_label_position("right")
ax4[ii].tick_params(which='both',
color='r',
labelsize=7,
labelcolor='m',
pad=0.1,
length=1,
rotation=0)
if ii < len(fdatasets) - 1:
ax4[ii].set_xlabel(' ')
else:
ax4[ii].set_xlabel('Time', rotation=0, fontsize=8, color='b')
if ii == 0:
ax4[ii].set_title(r + ' - Heading ' + roll_unit, fontsize=8)
leg4 = ax2[ii].legend(fontsize=6,
bbox_to_anchor=(0., 0.80, 1., .102),
loc=3,
ncol=3,
mode="expand",
borderaxespad=0.)
leg4._drawFrame = False
sfile = 'heading_plots'
save_file = os.path.join(save_dir, sfile)
fig4.savefig(str(save_file), dpi=150)
# velocity variable
u_name = [
u_var for u_var in sci_var if 'eastward_velocity' in u_var
]
v_name = [
v_var for v_var in sci_var if 'northward_velocity' in v_var
]
w_name = [w_var for w_var in sci_var if 'upward_velocity' in w_var]
w = ds[w_name[0]].values
w_unit = ds[w_name[0]].units
u = ds[u_name[0]].values
v = ds[v_name[0]].values
uv_magnitude = np.sqrt(u**2 + v**2)
uv_maxmag = max(uv_magnitude)
# 1D Quiver plot
ax[ii].quiver(time,
0,
u,
v,
color='r',
units='y',
scale_units='y',
scale=1,
headlength=1,
headaxislength=1,
width=0.004,
alpha=0.5)
u_fit = u[tilt_ind]
v_fit = v[tilt_ind]
ax[ii].quiver(time[tilt_ind],
0,
u_fit,
v_fit,
color='b',
units='y',
scale_units='y',
scale=1,
headlength=1,
headaxislength=1,
width=0.004,
alpha=0.5)
percent_bad = round(((len(u) - len(u_fit)) / len(u)) * 100, 2)
print(len(u_fit), len(u), percent_bad)
ax[ii].text(time[-1],
0,
' ' + str(percent_bad) + '%',
fontsize=5,
style='italic',
color='blue')
ax[ii].set_ylim(-uv_maxmag, uv_maxmag)
ax[ii].set_ylabel(str(deployment),
rotation=0,
fontsize=8,
color='b',
labelpad=11)
ax[ii].yaxis.set_label_position("right")
ax[ii].tick_params(which='both',
color='r',
labelsize=7,
labelcolor='m',
pad=0.1,
length=1,
rotation=0)
if ii < len(fdatasets) - 1:
ax[ii].set_xlabel(' ')
else:
ax[ii].set_xlabel('Time', rotation=0, fontsize=8, color='b')
if ii == 0:
ax[ii].set_title(
r + ' - Current Velocity ' + w_unit + '\n' +
' Currents in blue when pitch or roll are > 20 degrees',
fontsize=8)
# ax[ii].text(time[0], uv_magnitude- 0.05, 'mim: ' + str(round(min(uv_magnitude),3)) + ' , max: ' + str(round(max(uv_magnitude),3)), fontsize=8)
sfile = 'current_plot'
save_file = os.path.join(save_dir, sfile)
fig.savefig(str(save_file), dpi=150, bbox_inches='tight')
def main(sDir, url_list, start_time, end_time, preferred_only):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list and 'PRESF' in rd:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
for ud in udatasets: # filter out collocated data files
if 'PRESF' in ud.split('/')[-1]:
datasets.append(ud)
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
fdatasets = np.unique(fdatasets).tolist()
for fd in fdatasets:
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'.
format(start_time, end_time))
continue
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(
fd)
sci_vars = cf.return_science_vars(stream)
print('\nPlotting {} {}'.format(r, deployment))
array = subsite[0:2]
filename = '_'.join(fname.split('_')[:-1])
save_dir = os.path.join(sDir, array, subsite, refdes,
'timeseries_plots', deployment)
cf.create_dir(save_dir)
tm = ds['time'].values
t0 = pd.to_datetime(tm.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(tm.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes, method))
for var in sci_vars:
print(var)
if var != 'id':
#if var == 'presf_wave_burst_pressure':
y = ds[var]
fv = y._FillValue
if len(y.dims) == 1:
# Check if the array is all NaNs
if sum(np.isnan(y.values)) == len(y.values):
print('Array of all NaNs - skipping plot.')
# Check if the array is all fill values
elif len(y[y != fv]) == 0:
print('Array of all fill values - skipping plot.')
else:
# reject fill values
ind = y.values != fv
t = tm[ind]
y = y[ind]
# Plot all data
fig, ax = pf.plot_timeseries(t,
y,
y.name,
stdev=None)
ax.set_title((title + '\n' + t0 + ' - ' + t1),
fontsize=9)
sfile = '-'.join((filename, y.name, t0[:10]))
pf.save_fig(save_dir, sfile)
# Plot data with outliers removed
fig, ax = pf.plot_timeseries(t, y, y.name, stdev=5)
ax.set_title((title + '\n' + t0 + ' - ' + t1),
fontsize=9)
sfile = '-'.join(
(filename, y.name, t0[:10])) + '_rmoutliers'
pf.save_fig(save_dir, sfile)
else:
v = y.values.T
n_nan = np.sum(np.isnan(v))
# convert fill values to nans
try:
v[v == fv] = np.nan
except ValueError:
v = v.astype(float)
v[v == fv] = np.nan
n_fv = np.sum(np.isnan(v)) - n_nan
# plot before global ranges are removed
fig, ax = pf.plot_presf_2d(tm, v, y.name, y.units)
ax.set_title((title + '\n' + t0 + ' - ' + t1),
fontsize=9)
sfile = '-'.join((filename, var, t0[:10]))
pf.save_fig(save_dir, sfile)
# reject data outside of global ranges
[g_min, g_max] = cf.get_global_ranges(r, var)
if g_min is not None and g_max is not None:
v[v < g_min] = np.nan
v[v > g_max] = np.nan
n_grange = np.sum(np.isnan(v)) - n_fv - n_nan
if n_grange > 0:
# don't plot if the array is all nans
if len(np.unique(
np.isnan(v))) == 1 and np.unique(
np.isnan(v))[0] == True:
continue
else:
# plot after global ranges are removed
fig, ax = pf.plot_presf_2d(
tm, v, y.name, y.units)
title2 = 'removed: {} global ranges [{}, {}]'.format(
n_grange, g_min, g_max)
ax.set_title((title + '\n' + t0 + ' - ' +
t1 + '\n' + title2),
fontsize=9)
sfile = '-'.join(
(filename, var, t0[:10], 'rmgr'))
pf.save_fig(save_dir, sfile)
def main(sDir, ncdir):
rd_list = [ncdir.split('/')[-2]]
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
ps_df, n_streams = cf.get_preferred_stream_info(r)
# get end times of deployments
dr_data = cf.refdes_datareview_json(r)
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# filter datasets
fdatasets = []
for root, dirs, files in os.walk(ncdir):
for f in files:
if f.endswith('.nc'):
fdatasets.append(f)
# for u in url_list:
# splitter = u.split('/')[-2].split('-')
# rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
# if rd_check == r:
# udatasets = cf.get_nc_urls([u])
# datasets.append(udatasets)
# datasets = list(itertools.chain(*datasets))
# main_sensor = r.split('-')[-1]
# fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
methodstream = []
for f in fdatasets:
strm = '_'.join((f.split('-')[-2].split('_')[0], f.split('-')[-2].split('_')[1]))
methodstream.append('-'.join((f.split('-')[-3], strm)))
for ms in np.unique(methodstream):
fdatasets_sel = [x for x in fdatasets if ms in x]
save_dir = os.path.join(sDir, array, subsite, r, 'timeseries_plots_all')
cf.create_dir(save_dir)
stream_sci_vars_dict = dict()
for x in dr_data['instrument']['data_streams']:
dr_ms = '-'.join((x['method'], x['stream_name']))
if ms == dr_ms:
stream_sci_vars_dict[dr_ms] = dict(vars=dict())
sci_vars = dict()
for y in x['stream']['parameters']:
if y['data_product_type'] == 'Science Data':
sci_vars.update({y['name']: dict(db_units=y['unit'])})
if len(sci_vars) > 0:
stream_sci_vars_dict[dr_ms]['vars'] = sci_vars
sci_vars_dict = cd.initialize_empty_arrays(stream_sci_vars_dict, ms)
print('\nAppending data from files: {}'.format(ms))
for fd in fdatasets_sel:
ds = xr.open_dataset(os.path.join(ncdir, fd), mask_and_scale=False)
for var in list(sci_vars_dict[ms]['vars'].keys()):
sh = sci_vars_dict[ms]['vars'][var]
if ds[var].units == sh['db_units']:
if ds[var]._FillValue not in sh['fv']:
sh['fv'].append(ds[var]._FillValue)
if ds[var].units not in sh['units']:
sh['units'].append(ds[var].units)
tD = ds['time'].values
varD = ds[var].values
sh['t'] = np.append(sh['t'], tD)
sh['values'] = np.append(sh['values'], varD)
print('\nPlotting data')
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(sv)
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
t0 = pd.to_datetime(min(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
x = vinfo['t']
y = vinfo['values']
# reject NaNs
nan_ind = ~np.isnan(y)
x_nonan = x[nan_ind]
y_nonan = y[nan_ind]
# reject fill values
fv_ind = y_nonan != vinfo['fv'][0]
x_nonan_nofv = x_nonan[fv_ind]
y_nonan_nofv = y_nonan[fv_ind]
# reject extreme values
Ev_ind = cf.reject_extreme_values(y_nonan_nofv)
y_nonan_nofv_nE = y_nonan_nofv[Ev_ind]
x_nonan_nofv_nE = x_nonan_nofv[Ev_ind]
# reject values outside global ranges:
global_min, global_max = cf.get_global_ranges(r, sv)
if global_min is not None and global_max is not None:
gr_ind = cf.reject_global_ranges(y_nonan_nofv_nE, global_min, global_max)
y_nonan_nofv_nE_nogr = y_nonan_nofv_nE[gr_ind]
x_nonan_nofv_nE_nogr = x_nonan_nofv_nE[gr_ind]
else:
y_nonan_nofv_nE_nogr = y_nonan_nofv_nE
x_nonan_nofv_nE_nogr = x_nonan_nofv_nE
title = ' '.join((r, ms.split('-')[0]))
if len(y_nonan_nofv) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((r, sv))
else:
sname = '-'.join((r, m, sv))
# Plot all data
fig, ax = pf.plot_timeseries_all(x_nonan_nofv, y_nonan_nofv, sv, sv_units, stdev=None)
ax.set_title((title + '\nDeployments: ' + str(sorted(deployments)) + '\n' + t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes, color='b', linestyle='--', linewidth=.6)
# if global_min is not None and global_max is not None:
# ax.axhline(y=global_min, color='r', linestyle='--', linewidth=.6)
# ax.axhline(y=global_max, color='r', linestyle='--', linewidth=.6)
pf.save_fig(save_dir, sname)
# Plot data with extreme values, data outside global ranges and outliers removed
fig, ax = pf.plot_timeseries_all(x_nonan_nofv_nE_nogr, y_nonan_nofv_nE_nogr, sv, sv_units, stdev=5)
ax.set_title((title + '\nDeployments: ' + str(sorted(deployments)) + '\n' + t0 + ' - ' + t1),
fontsize=8)
for etimes in end_times:
ax.axvline(x=etimes, color='b', linestyle='--', linewidth=.6)
# if global_min is not None and global_max is not None:
# ax.axhline(y=global_min, color='r', linestyle='--', linewidth=.6)
# ax.axhline(y=global_max, color='r', linestyle='--', linewidth=.6)
sfile = '_'.join((sname, 'rmoutliers'))
pf.save_fig(save_dir, sfile)
def main(url_list, sDir, deployment_num, start_time, end_time, preferred_only,
n_std, surface_params, depth_params):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list and 'ENG' not in rd:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(
main_sensor, fdatasets)
for fd in fdatasets_sel:
part_d = fd.split('/')[-1]
print('\n{}'.format(part_d))
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(
fd)
array = subsite[0:2]
sci_vars = cf.return_science_vars(stream)
if 'CE05MOAS' in r or 'CP05MOAS' in r: # for coastal gliders, get m_water_depth for bathymetry
eng = '-'.join((r.split('-')[0], r.split('-')[1],
'00-ENG000000', method, 'glider_eng'))
eng_url = [s for s in url_list if eng in s]
if len(eng_url) == 1:
eng_datasets = cf.get_nc_urls(eng_url)
# filter out collocated datasets
eng_dataset = [
j for j in eng_datasets
if (eng in j.split('/')[-1]
and deployment in j.split('/')[-1])
]
if len(eng_dataset) > 0:
ds_eng = xr.open_dataset(eng_dataset[0],
mask_and_scale=False)
t_eng = ds_eng['time'].values
m_water_depth = ds_eng['m_water_depth'].values
# m_altimeter_status = 0 means a good reading (not nan or -1)
eng_ind = ds_eng['m_altimeter_status'].values == 0
m_water_depth = m_water_depth[eng_ind]
t_eng = t_eng[eng_ind]
else:
print('No engineering file for deployment {}'.format(
deployment))
m_water_depth = None
t_eng = None
else:
m_water_depth = None
t_eng = None
else:
m_water_depth = None
t_eng = None
if deployment_num is not None:
if int(deployment.split('0')[-1]) is not deployment_num:
print(type(int(deployment.split('0')[-1])),
type(deployment_num))
continue
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'.
format(start_time, end_time))
continue
stime = start_time.strftime('%Y-%m-%d')
etime = end_time.strftime('%Y-%m-%d')
ext = stime + 'to' + etime # .join((ds0_method, ds1_method
save_dir_profile = os.path.join(sDir, array, subsite, refdes,
'profile_plots', deployment,
ext)
save_dir_xsection = os.path.join(sDir, array, subsite, refdes,
'xsection_plots', deployment,
ext)
save_dir_4d = os.path.join(sDir, array, subsite, refdes,
'xsection_plots_4d', deployment,
ext)
else:
save_dir_profile = os.path.join(sDir, array, subsite, refdes,
'profile_plots', deployment)
save_dir_xsection = os.path.join(sDir, array, subsite, refdes,
'xsection_plots', deployment)
save_dir_4d = os.path.join(sDir, array, subsite, refdes,
'xsection_plots_4d', deployment)
tm = ds['time'].values
try:
ds_lat = ds['lat'].values
except KeyError:
ds_lat = None
print('No latitude variable in file')
try:
ds_lon = ds['lon'].values
except KeyError:
ds_lon = None
print('No longitude variable in file')
# get pressure variable
y, y_units, press = cf.add_pressure_to_dictionary_of_sci_vars(ds)
for sv in sci_vars:
print(sv)
if 'pressure' not in sv:
z = ds[sv].values
fv = ds[sv]._FillValue
sv_units = ds[sv].units
# Check if the array is all NaNs
if sum(np.isnan(z)) == len(z):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z[z != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
# reject erroneous data
dtime, zpressure, ndata, lenfv, lennan, lenev, lengr, global_min, global_max, lat, lon = \
cf.reject_erroneous_data(r, sv, tm, y, z, fv, ds_lat, ds_lon)
# get rid of 0.0 data
if 'CTD' in r:
ind = zpressure > 0.0
else:
ind = ndata > 0.0
lenzero = np.sum(~ind)
dtime = dtime[ind]
zpressure = zpressure[ind]
ndata = ndata[ind]
if ds_lat is not None and ds_lon is not None:
lat = lat[ind]
lon = lon[ind]
else:
lat = None
lon = None
t0 = pd.to_datetime(
dtime.min()).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(
dtime.max()).strftime('%Y-%m-%dT%H:%M:%S')
title = ' '.join((deployment, refdes,
method)) + '\n' + t0 + ' to ' + t1
# reject time range from data portal file export
t_portal, z_portal, y_portal, lat_portal, lon_portal = \
cf.reject_timestamps_dataportal(subsite, r, dtime, zpressure, ndata, lat, lon)
print(
'removed {} data points using visual inspection of data'
.format(len(ndata) - len(z_portal)))
# create data groups
columns = ['tsec', 'dbar', str(sv)]
# min_r = int(round(min(y_portal) - zcell_size))
# max_r = int(round(max(y_portal) + zcell_size))
# ranges = list(range(min_r, max_r, zcell_size))
#ranges = [0, 10, 20, 30, 40, 50, 60, 70, 80, 200]
range1 = list(
range(surface_params[0], surface_params[1],
surface_params[2]))
range2 = list(
range(depth_params[0],
depth_params[1] + depth_params[2],
depth_params[2]))
ranges = range1 + range2
groups, d_groups = gt.group_by_depth_range(
t_portal, y_portal, z_portal, columns, ranges)
if 'scatter' in sv:
n_std = None # to use percentile
else:
n_std = n_std
# get percentile analysis for printing on the profile plot
inpercentile = [surface_params[3]] * len(
range1) + [depth_params[3]] * len(range2)
n_std = [surface_params[3]] * len(
range1) + [depth_params[3]] * len(range2)
y_plt, n_med, n_min, n_max, n0_std, n1_std, l_arr, time_ex = reject_timestamps_in_groups(
groups, d_groups, n_std, inpercentile)
"""
Plot all data
"""
if len(tm) > 0:
cf.create_dir(save_dir_profile)
cf.create_dir(save_dir_xsection)
sname = '-'.join((r, method, sv))
sfileall = '_'.join(('all_data', sname))
'''
profile plot
'''
xlabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
clabel = 'Time'
fig, ax = pf.plot_profiles(z,
y,
tm,
ylabel,
xlabel,
clabel,
stdev=None)
ax.set_title(title, fontsize=9)
fig.tight_layout()
pf.save_fig(save_dir_profile, sfileall)
'''
xsection plot
'''
clabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
fig, ax, bar = pf.plot_xsection(subsite,
tm,
y,
z,
clabel,
ylabel,
t_eng,
m_water_depth,
inpercentile=None,
stdev=None)
ax.set_title(title, fontsize=9)
fig.tight_layout()
pf.save_fig(save_dir_xsection, sfileall)
"""
Plot cleaned-up data
"""
if len(dtime) > 0:
sfile = '_'.join(('rm_erroneous_data', sname))
'''
profile plot
'''
xlabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
clabel = 'Time'
fig, ax = pf.plot_profiles(z_portal,
y_portal,
t_portal,
ylabel,
xlabel,
clabel,
stdev=None)
ax.set_title(title, fontsize=9)
ax.plot(n_med, y_plt, '.k')
ax.fill_betweenx(y_plt,
n0_std,
n1_std,
color='m',
alpha=0.2)
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} zeros'.format(lenfv, lennan, lenev, lengr,
global_min, global_max,
lenzero) +
'\nexcluded {} suspect data points when inspected visually'
.format(len(ndata) - len(z_portal)) +
'\n(black) data median in {} dbar segments (break at {} dbar)'
.format([surface_params[2], depth_params[2]],
depth_params[0]) +
'\n(magenta) upper and lower {} percentile envelope in {} dbar segments'
.format(
[surface_params[3], depth_params[3]],
[surface_params[2], depth_params[2]]), )
ax.legend(leg_text,
loc='upper center',
bbox_to_anchor=(0.5, -0.17),
fontsize=6)
fig.tight_layout()
pf.save_fig(save_dir_profile, sfile)
'''
xsection plot
'''
clabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[0] + ")"
# plot non-erroneous data
fig, ax, bar = pf.plot_xsection(subsite,
t_portal,
y_portal,
z_portal,
clabel,
ylabel,
t_eng,
m_water_depth,
inpercentile=None,
stdev=None)
ax.set_title(title, fontsize=9)
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} zeros'.format(lenfv, lennan, lenev, lengr,
global_min, global_max,
lenzero) +
'\nexcluded {} suspect data points when inspected visually'
.format(len(ndata) - len(z_portal)), )
ax.legend(leg_text,
loc='upper center',
bbox_to_anchor=(0.5, -0.17),
fontsize=6)
fig.tight_layout()
pf.save_fig(save_dir_xsection, sfile)
'''
4D plot for gliders only
'''
if 'MOAS' in r:
if ds_lat is not None and ds_lon is not None:
cf.create_dir(save_dir_4d)
clabel = sv + " (" + sv_units + ")"
zlabel = press[0] + " (" + y_units[0] + ")"
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
sct = ax.scatter(lon_portal,
lat_portal,
y_portal,
c=z_portal,
s=2)
cbar = plt.colorbar(sct,
label=clabel,
extend='both')
cbar.ax.tick_params(labelsize=8)
ax.invert_zaxis()
ax.view_init(25, 32)
ax.invert_xaxis()
ax.invert_yaxis()
ax.set_zlabel(zlabel, fontsize=9)
ax.set_ylabel('Latitude', fontsize=9)
ax.set_xlabel('Longitude', fontsize=9)
ax.set_title(title, fontsize=9)
pf.save_fig(save_dir_4d, sfile)
def main(url_list, sDir, deployment_num, start_time, end_time, preferred_only,
zdbar, n_std, inpercentile, zcell_size):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
if rd not in rd_list and 'ENG' not in rd and 'ADCP' not in rd:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets_sel = cf.filter_collocated_instruments(
main_sensor, fdatasets)
for fd in fdatasets_sel:
part_d = fd.split('/')[-1]
print('\n{}'.format(part_d))
ds = xr.open_dataset(fd, mask_and_scale=False)
ds = ds.swap_dims({'obs': 'time'})
fname, subsite, refdes, method, stream, deployment = cf.nc_attributes(
fd)
array = subsite[0:2]
sci_vars = cf.return_science_vars(stream)
# if 'CE05MOAS' in r or 'CP05MOAS' in r: # for coastal gliders, get m_water_depth for bathymetry
# eng = '-'.join((r.split('-')[0], r.split('-')[1], '00-ENG000000', method, 'glider_eng'))
# eng_url = [s for s in url_list if eng in s]
# if len(eng_url) == 1:
# eng_datasets = cf.get_nc_urls(eng_url)
# # filter out collocated datasets
# eng_dataset = [j for j in eng_datasets if (eng in j.split('/')[-1] and deployment in j.split('/')[-1])]
# if len(eng_dataset) > 0:
# ds_eng = xr.open_dataset(eng_dataset[0], mask_and_scale=False)
# t_eng = ds_eng['time'].values
# m_water_depth = ds_eng['m_water_depth'].values
#
# # m_altimeter_status = 0 means a good reading (not nan or -1)
# try:
# eng_ind = ds_eng['m_altimeter_status'].values == 0
# except KeyError:
# eng_ind = (~np.isnan(m_water_depth)) & (m_water_depth >= 0)
#
# m_water_depth = m_water_depth[eng_ind]
# t_eng = t_eng[eng_ind]
#
# # get rid of any remaining nans or fill values
# eng_ind2 = (~np.isnan(m_water_depth)) & (m_water_depth >= 0)
# m_water_depth = m_water_depth[eng_ind2]
# t_eng = t_eng[eng_ind2]
# else:
# print('No engineering file for deployment {}'.format(deployment))
# m_water_depth = None
# t_eng = None
# else:
# m_water_depth = None
# t_eng = None
# else:
# m_water_depth = None
# t_eng = None
if deployment_num is not None:
if int(int(deployment[-4:])) is not deployment_num:
print(type(int(deployment[-4:])), type(deployment_num))
continue
if start_time is not None and end_time is not None:
ds = ds.sel(time=slice(start_time, end_time))
if len(ds['time'].values) == 0:
print(
'No data to plot for specified time range: ({} to {})'.
format(start_time, end_time))
continue
stime = start_time.strftime('%Y-%m-%d')
etime = end_time.strftime('%Y-%m-%d')
ext = stime + 'to' + etime # .join((ds0_method, ds1_method
save_dir_profile = os.path.join(sDir, array, subsite, refdes,
'profile_plots', deployment,
ext)
save_dir_xsection = os.path.join(sDir, array, subsite, refdes,
'xsection_plots', deployment,
ext)
save_dir_4d = os.path.join(sDir, array, subsite, refdes,
'xsection_plots_4d', deployment,
ext)
else:
save_dir_profile = os.path.join(sDir, array, subsite, refdes,
'profile_plots', deployment)
save_dir_xsection = os.path.join(sDir, array, subsite, refdes,
'xsection_plots', deployment)
save_dir_4d = os.path.join(sDir, array, subsite, refdes,
'xsection_plots_4d', deployment)
texclude_dir = os.path.join(sDir, array, subsite, refdes,
'time_to_exclude')
cf.create_dir(texclude_dir)
time1 = ds['time'].values
try:
ds_lat1 = ds['lat'].values
except KeyError:
ds_lat1 = None
print('No latitude variable in file')
try:
ds_lon1 = ds['lon'].values
except KeyError:
ds_lon1 = None
print('No longitude variable in file')
# get pressure variable
pvarname, y1, y_units, press, y_fillvalue = cf.add_pressure_to_dictionary_of_sci_vars(
ds)
# prepare file to list timestamps with suspect data for each data parameter
stat_data = pd.DataFrame(
columns=['deployments', 'time_to_exclude'])
file_exclude = '{}/{}_{}_{}_excluded_timestamps.csv'.format(
texclude_dir, deployment, refdes, method)
stat_data.to_csv(file_exclude, index=True)
# loop through sensor-data parameters
for sv in sci_vars:
print(sv)
if 'pressure' not in sv:
z1 = ds[sv].values
fv = ds[sv]._FillValue
sv_units = ds[sv].units
# Check if the array is all NaNs
if sum(np.isnan(z1)) == len(z1):
print('Array of all NaNs - skipping plot.')
continue
# Check if the array is all fill values
elif len(z1[z1 != fv]) == 0:
print('Array of all fill values - skipping plot.')
continue
else:
# remove unreasonable pressure data (e.g. for surface piercing profilers)
if zdbar:
po_ind = (0 < y1) & (y1 < zdbar)
n_zdbar = np.sum(~po_ind)
tm = time1[po_ind]
y = y1[po_ind]
z = z1[po_ind]
ds_lat = ds_lat1[po_ind]
ds_lon = ds_lon1[po_ind]
print('{} in water depth > {} dbar'.format(
n_zdbar, zdbar))
else:
tm = time1
y = y1
z = z1
ds_lat = ds_lat1
ds_lon = ds_lon1
# reject erroneous data
dtime, zpressure, ndata, lenfv, lennan, lenev, lengr, global_min, global_max, lat, lon = \
cf.reject_erroneous_data(r, sv, tm, y, z, fv, ds_lat, ds_lon)
# get rid of 0.0 data
if sv == 'salinity':
ind = ndata > 30
elif sv == 'density':
ind = ndata > 1022.5
elif sv == 'conductivity':
ind = ndata > 3.45
else:
ind = ndata > 0
# if sv == 'sci_flbbcd_chlor_units':
# ind = ndata < 7.5
# elif sv == 'sci_flbbcd_cdom_units':
# ind = ndata < 25
# else:
# ind = ndata > 0.0
# if 'CTD' in r:
# ind = zpressure > 0.0
# else:
# ind = ndata > 0.0
lenzero = np.sum(~ind)
dtime = dtime[ind]
zpressure = zpressure[ind]
ndata = ndata[ind]
if ds_lat is not None and ds_lon is not None:
lat = lat[ind]
lon = lon[ind]
else:
lat = None
lon = None
if len(dtime) > 0:
# reject time range from data portal file export
t_portal, z_portal, y_portal, lat_portal, lon_portal = \
cf.reject_timestamps_dataportal(subsite, r, dtime, zpressure, ndata, lat, lon)
print(
'removed {} data points using visual inspection of data'
.format(len(ndata) - len(z_portal)))
# create data groups
if len(y_portal) > 0:
columns = ['tsec', 'dbar', str(sv)]
min_r = int(round(min(y_portal) - zcell_size))
max_r = int(round(max(y_portal) + zcell_size))
ranges = list(range(min_r, max_r, zcell_size))
groups, d_groups = gt.group_by_depth_range(
t_portal, y_portal, z_portal, columns,
ranges)
if 'scatter' in sv:
n_std = None # to use percentile
else:
n_std = n_std
# identifying timestamps from percentile analysis
y_avg, n_avg, n_min, n_max, n0_std, n1_std, l_arr, time_ex = cf.reject_timestamps_in_groups(
groups, d_groups, n_std, inpercentile)
"""
writing timestamps to .csv file to use with data_range.py script
"""
if len(time_ex) != 0:
t_exclude = time_ex[0]
for i in range(
len(time_ex))[1:len(time_ex)]:
t_exclude = '{}, {}'.format(
t_exclude, time_ex[i])
stat_data = pd.DataFrame(
{
'deployments': deployment,
'time_to_exclude': t_exclude
},
index=[sv])
stat_data.to_csv(file_exclude,
index=True,
mode='a',
header=False)
# rejecting timestamps from percentile analysis
if len(time_ex) > 0:
t_nospct, z_nospct, y_nospct = cf.reject_suspect_data(
t_portal, y_portal, z_portal, time_ex)
else:
t_nospct = t_portal
z_nospct = z_portal
y_nospct = y_portal
"""
Plot data
"""
if len(t_nospct) > 0:
if len(t_nospct) != len(dtime):
cf.create_dir(save_dir_profile)
cf.create_dir(save_dir_xsection)
sname = '-'.join((r, method, sv))
sfile = '_'.join(
('rm_suspect_data', sname,
pd.to_datetime(
t_nospct.min()).strftime(
'%Y%m%d')))
t0 = pd.to_datetime(
t_nospct.min()).strftime(
'%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(
t_nospct.max()).strftime(
'%Y-%m-%dT%H:%M:%S')
title = ' '.join(
(deployment, refdes,
method)) + '\n' + t0 + ' to ' + t1
if zdbar:
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges '
'[{} - {}], {} unreasonable values'
.format(
lenfv, lennan, lenev,
lengr, global_min,
global_max, lenzero) +
'\nremoved {} in the upper and lower {} percentile of data grouped in {} '
'dbar segments'.format(
len(z_portal) -
len(z_nospct),
inpercentile, zcell_size) +
'\nexcluded {} suspect data points when inspected visually'
.format(
len(ndata) - len(z_portal))
+
'\nexcluded {} suspect data in water depth greater than {} dbar'
.format(n_zdbar, zdbar), )
elif n_std:
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} unreasonable values'.
format(lenfv, lennan, lenev,
lengr, global_min,
global_max, lenzero) +
'\nremoved {} data points +/- {} SD of data grouped in {} dbar segments'
.format(
len(z_portal) -
len(z_nospct), n_std,
zcell_size) +
'\nexcluded {} suspect data points when inspected visually'
.format(
len(ndata) -
len(z_portal)), )
else:
leg_text = (
'removed {} fill values, {} NaNs, {} Extreme Values (1e7), {} Global ranges [{} - {}], '
'{} unreasonable values'.
format(lenfv, lennan, lenev,
lengr, global_min,
global_max, lenzero) +
'\nremoved {} in the upper and lower {} percentile of data grouped in {} dbar segments'
.format(
len(z_portal) -
len(z_nospct),
inpercentile, zcell_size) +
'\nexcluded {} suspect data points when inspected visually'
.format(
len(ndata) -
len(z_portal)), )
'''
profile plot
'''
xlabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[
0] + ")"
clabel = 'Time'
# plot non-erroneous data
print('plotting profile')
fig, ax = pf.plot_profiles(z_nospct,
y_nospct,
t_nospct,
ylabel,
xlabel,
clabel,
stdev=None)
ax.set_title(title, fontsize=9)
ax.plot(n_avg, y_avg, '-k')
#ax.fill_betweenx(y_avg, n0_std, n1_std, color='m', alpha=0.2)
ax.legend(leg_text,
loc='upper center',
bbox_to_anchor=(0.5, -0.17),
fontsize=6)
fig.tight_layout()
pf.save_fig(save_dir_profile, sfile)
'''
xsection plot
'''
print('plotting xsection')
clabel = sv + " (" + sv_units + ")"
ylabel = press[0] + " (" + y_units[
0] + ")"
# plot bathymetry only within data time ranges
# if t_eng is not None:
# eng_ind = (t_eng >= np.nanmin(t_array)) & (t_eng <= np.nanmax(t_array))
# t_eng = t_eng[eng_ind]
# m_water_depth = m_water_depth[eng_ind]
# plot non-erroneous data
fig, ax, bar = pf.plot_xsection(
subsite,
t_nospct,
y_nospct,
z_nospct,
clabel,
ylabel,
t_eng=None,
m_water_depth=None,
inpercentile=inpercentile,
stdev=None)
ax.set_title(title, fontsize=9)
ax.legend(leg_text,
loc='upper center',
bbox_to_anchor=(0.5, -0.17),
fontsize=6)
fig.tight_layout()
pf.save_fig(save_dir_xsection, sfile)
def main(sDir, url_list, preferred_only):
rd_list = []
ms_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
if ms not in ms_list:
ms_list.append(ms)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
# filter datasets
datasets = []
for u in url_list:
print(u)
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
fdatasets = []
if preferred_only == 'yes':
# get the preferred stream information
ps_df, n_streams = cf.get_preferred_stream_info(r)
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, fdatasets)
# ps_df, n_streams = cf.get_preferred_stream_info(r)
# get end times of deployments
dr_data = cf.refdes_datareview_json(r)
deployments = []
end_times = []
for index, row in ps_df.iterrows():
deploy = row['deployment']
deploy_info = get_deployment_information(dr_data, int(deploy[-4:]))
deployments.append(int(deploy[-4:]))
end_times.append(pd.to_datetime(deploy_info['stop_date']))
# # filter datasets
# datasets = []
# for u in url_list:
# print(u)
# splitter = u.split('/')[-2].split('-')
# rd_check = '-'.join((splitter[1], splitter[2], splitter[3], splitter[4]))
# if rd_check == r:
# udatasets = cf.get_nc_urls([u])
# datasets.append(udatasets)
# datasets = list(itertools.chain(*datasets))
# main_sensor = r.split('-')[-1]
# fdatasets = cf.filter_collocated_instruments(main_sensor, datasets)
# fdatasets = cf.filter_other_streams(r, ms_list, fdatasets)
methodstream = []
for f in fdatasets:
methodstream.append('-'.join((f.split('/')[-2].split('-')[-2],
f.split('/')[-2].split('-')[-1])))
ms_dict = save_dir_path(ms_list)
for ms in np.unique(methodstream):
fdatasets_sel = [x for x in fdatasets if ms in x]
check_ms = ms.split('-')[1]
if 'recovered' in check_ms:
check_ms = check_ms.split('_recovered')[0]
if ms_dict['ms_count'][ms_dict['ms_unique'] == ms.split('-')
[0]] == 1:
save_dir = os.path.join(sDir, array, subsite, r,
'timeseries_yearly_plot',
ms.split('-')[0])
else:
save_dir = os.path.join(sDir, array, subsite, r,
'timeseries_yearly_plot',
ms.split('-')[0], check_ms)
cf.create_dir(save_dir)
stream_sci_vars_dict = dict()
for x in dr_data['instrument']['data_streams']:
dr_ms = '-'.join((x['method'], x['stream_name']))
if ms == dr_ms:
stream_sci_vars_dict[dr_ms] = dict(vars=dict())
sci_vars = dict()
for y in x['stream']['parameters']:
if y['data_product_type'] == 'Science Data':
sci_vars.update(
{y['name']: dict(db_units=y['unit'])})
if len(sci_vars) > 0:
stream_sci_vars_dict[dr_ms]['vars'] = sci_vars
sci_vars_dict = cd.initialize_empty_arrays(stream_sci_vars_dict,
ms)
print('\nAppending data from files: {}'.format(ms))
for fd in fdatasets_sel:
ds = xr.open_dataset(fd, mask_and_scale=False)
print(fd)
for var in list(sci_vars_dict[ms]['vars'].keys()):
sh = sci_vars_dict[ms]['vars'][var]
try:
ds[var]
print(var)
deployment_num = np.unique(ds['deployment'].values)[0]
sh['deployments'] = np.append(sh['deployments'],
deployment_num)
if ds[var].units == sh['db_units']:
if ds[var]._FillValue not in sh['fv']:
sh['fv'].append(ds[var]._FillValue)
if ds[var].units not in sh['units']:
sh['units'].append(ds[var].units)
tD = ds['time'].values
varD = ds[var].values
sh['t'] = np.append(sh['t'], tD)
sh['values'] = np.append(sh['values'], varD)
except KeyError:
print('KeyError: ', var)
print('\nPlotting data')
for m, n in sci_vars_dict.items():
for sv, vinfo in n['vars'].items():
print(sv)
if len(vinfo['t']) < 1:
print('no variable data to plot')
else:
sv_units = vinfo['units'][0]
deployments_num = vinfo['deployments']
fv = vinfo['fv'][0]
t0 = pd.to_datetime(min(
vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
t1 = pd.to_datetime(max(
vinfo['t'])).strftime('%Y-%m-%dT%H:%M:%S')
x = vinfo['t']
y = vinfo['values']
# reject NaNs
nan_ind = ~np.isnan(y)
x_nonan = x[nan_ind]
y_nonan = y[nan_ind]
# reject fill values
fv_ind = y_nonan != vinfo['fv'][0]
x_nonan_nofv = x_nonan[fv_ind]
y_nonan_nofv = y_nonan[fv_ind]
# reject extreme values
Ev_ind = cf.reject_extreme_values(y_nonan_nofv)
y_nonan_nofv_nE = y_nonan_nofv[Ev_ind]
x_nonan_nofv_nE = x_nonan_nofv[Ev_ind]
# reject values outside global ranges:
global_min, global_max = cf.get_global_ranges(r, sv)
print('global ranges: ', global_min, global_max)
if global_min and global_max:
gr_ind = cf.reject_global_ranges(
y_nonan_nofv_nE, global_min, global_max)
y_nonan_nofv_nE_nogr = y_nonan_nofv_nE[gr_ind]
x_nonan_nofv_nE_nogr = x_nonan_nofv_nE[gr_ind]
else:
y_nonan_nofv_nE_nogr = y_nonan_nofv_nE
x_nonan_nofv_nE_nogr = x_nonan_nofv_nE
# check array length
if len(y_nonan_nofv_nE_nogr) > 0:
if m == 'common_stream_placeholder':
sname = '-'.join((r, sv))
print(var, 'empty array')
else:
sname = '-'.join((r, m, sv))
# group data by year
groups, g_data = gt.group_by_time_range(
x_nonan_nofv_nE_nogr, y_nonan_nofv_nE_nogr,
'A')
# create bins
# groups_min = min(groups.describe()['DO']['min'])
# lower_bound = int(round(groups_min))
# groups_max = max(groups.describe()['DO']['max'])
# if groups_max < 1:
# upper_bound = 1
# step_bound = 1
# else:
# upper_bound = int(round(groups_max + (groups_max / 50)))
# step_bound = int(round((groups_max - groups_min) / 10))
#
# if step_bound == 0:
# step_bound += 1
#
# if (upper_bound - lower_bound) == step_bound:
# lower_bound -= 1
# upper_bound += 1
# if (upper_bound - lower_bound) < step_bound:
# print('<')
# step_bound = int(round(step_bound / 10))
# print(lower_bound, upper_bound, step_bound)
# bin_range = list(range(lower_bound, upper_bound, step_bound))
# print(bin_range)
# preparing color palette
colors = color_names[:len(groups)]
# colors = [color['color'] for color in
# list(pyplot.rcParams['axes.prop_cycle'][:len(groups)])]
fig0, ax0 = pyplot.subplots(nrows=2, ncols=1)
# subplot for histogram and basic statistics table
ax0[1].axis('off')
ax0[1].axis('tight')
the_table = ax0[1].table(
cellText=groups.describe().round(2).values,
rowLabels=groups.describe().index.year,
rowColours=colors,
colLabels=groups.describe().columns.levels[1],
loc='center')
the_table.set_fontsize(5)
# subplot for data
fig, ax = pyplot.subplots(nrows=len(groups),
ncols=1,
sharey=True)
if len(groups) == 1:
ax = [ax]
t = 1
for ny in range(len(groups)):
# prepare data for plotting
y_data = g_data[ny + (t + 1)].dropna(axis=0)
x_time = g_data[ny + t].dropna(axis=0)
t += 1
if len(y_data) != 0 and len(x_time) != 0:
n_year = x_time[0].year
col_name = str(n_year)
serie_n = pd.DataFrame(columns=[col_name],
index=x_time)
serie_n[col_name] = list(y_data[:])
# plot histogram
# serie_n.plot.hist(ax=ax0[0], bins=bin_range,
# histtype='bar', color=colors[ny], stacked=True)
if len(serie_n) != 1:
serie_n.plot.kde(ax=ax0[0],
color=colors[ny])
ax0[0].legend(fontsize=8,
bbox_to_anchor=(0., 1.12,
1.,
.102),
loc=3,
ncol=len(groups),
mode="expand",
borderaxespad=0.)
# ax0[0].set_xticks(bin_range)
ax0[0].set_xlabel('Observation Ranges',
fontsize=8)
ax0[0].set_ylabel(
'Density', fontsize=8
) #'Number of Observations'
ax0[0].set_title(
ms.split('-')[0] + ' (' + sv +
', ' + sv_units + ')' +
' Kernel Density Estimates',
fontsize=8)
# plot data
serie_n.plot(ax=ax[ny],
linestyle='None',
marker='.',
markersize=0.5,
color=colors[ny])
ax[ny].legend().set_visible(False)
# plot Mean and Standard deviation
ma = serie_n.rolling('86400s').mean()
mstd = serie_n.rolling('86400s').std()
ax[ny].plot(ma.index,
ma[col_name].values,
'k',
linewidth=0.15)
ax[ny].fill_between(
mstd.index,
ma[col_name].values -
2 * mstd[col_name].values,
ma[col_name].values +
2 * mstd[col_name].values,
color='b',
alpha=0.2)
# prepare the time axis parameters
datemin = datetime.date(n_year, 1, 1)
datemax = datetime.date(n_year, 12, 31)
ax[ny].set_xlim(datemin, datemax)
xlocator = mdates.MonthLocator(
) # every month
myFmt = mdates.DateFormatter('%m')
ax[ny].xaxis.set_minor_locator(
xlocator)
ax[ny].xaxis.set_major_formatter(myFmt)
# prepare the time axis parameters
# ax[ny].set_yticks(bin_range)
ylocator = MaxNLocator(prune='both',
nbins=3)
ax[ny].yaxis.set_major_locator(
ylocator)
# format figure
ax[ny].tick_params(axis='both',
color='r',
labelsize=7,
labelcolor='m')
if ny < len(groups) - 1:
ax[ny].tick_params(
which='both',
pad=0.1,
length=1,
labelbottom=False)
ax[ny].set_xlabel(' ')
else:
ax[ny].tick_params(which='both',
color='r',
labelsize=7,
labelcolor='m',
pad=0.1,
length=1,
rotation=0)
ax[ny].set_xlabel('Months',
rotation=0,
fontsize=8,
color='b')
ax[ny].set_ylabel(n_year,
rotation=0,
fontsize=8,
color='b',
labelpad=20)
ax[ny].yaxis.set_label_position(
"right")
if ny == 0:
if global_min and global_max:
ax[ny].set_title(
sv + '( ' + sv_units +
') -- Global Range: [' +
str(int(global_min)) +
',' +
str(int(global_max)) +
'] \n'
'Plotted: Data, Mean and 2STD (Method: One day rolling window calculations) \n',
fontsize=8)
else:
ax[ny].set_title(
sv + '( ' + sv_units +
') -- Global Range: [] \n'
'Plotted: Data, Mean and 2STD (Method: One day rolling window calculations) \n',
fontsize=8)
# plot global ranges
# ax[ny].axhline(y=global_min, color='r', linestyle='--', linewidth=.6)
# ax[ny].axhline(y=global_max, color='r', linestyle='--', linewidth=.6)
# mark deployment end times on figure
ymin, ymax = ax[ny].get_ylim()
#dep = 1
for etimes in range(len(end_times)):
if end_times[
etimes].year == n_year:
ax[ny].axvline(
x=end_times[etimes],
color='b',
linestyle='--',
linewidth=.6)
ax[ny].text(
end_times[etimes],
ymin,
'End' +
str(deployments_num[etimes]
),
fontsize=6,
style='italic',
bbox=dict(boxstyle='round',
ec=(0., 0.5,
0.5),
fc=(1., 1., 1.)))
# dep += 1
# ax[ny].set_ylim(5, 12)
# save figure to a file
sfile = '_'.join(('all', sname))
save_file = os.path.join(save_dir, sfile)
fig.savefig(str(save_file), dpi=150)
sfile = '_'.join(('Statistics', sname))
save_file = os.path.join(save_dir, sfile)
fig0.savefig(str(save_file), dpi=150)
pyplot.close()
def main(sDir, url_list, preferred_only, name_list):
rd_list = []
for uu in url_list:
elements = uu.split('/')[-2].split('-')
rd = '-'.join((elements[1], elements[2], elements[3], elements[4]))
ms = uu.split(rd + '-')[1].split('/')[0]
if rd not in rd_list:
rd_list.append(rd)
for r in rd_list:
print('\n{}'.format(r))
subsite = r.split('-')[0]
array = subsite[0:2]
datasets = []
for u in url_list:
splitter = u.split('/')[-2].split('-')
rd_check = '-'.join(
(splitter[1], splitter[2], splitter[3], splitter[4]))
if rd_check == r:
udatasets = cf.get_nc_urls([u])
datasets.append(udatasets)
datasets = list(itertools.chain(*datasets))
if preferred_only == 'yes':
ps_df, n_streams = cf.get_preferred_stream_info(r)
fdatasets = []
for index, row in ps_df.iterrows():
for ii in range(n_streams):
try:
rms = '-'.join((r, row[ii]))
except TypeError:
continue
for dd in datasets:
spl = dd.split('/')[-2].split('-')
catalog_rms = '-'.join(
(spl[1], spl[2], spl[3], spl[4], spl[5], spl[6]))
fdeploy = dd.split('/')[-1].split('_')[0]
if rms == catalog_rms and fdeploy == row['deployment']:
fdatasets.append(dd)
else:
fdatasets = datasets
main_sensor = r.split('-')[-1]
fdatasets = cf.filter_collocated_instruments(main_sensor, fdatasets)
vars_df = compare_variable_attributes(fdatasets, r)
if len(np.unique(vars_df.index.values)) == 1 and len(
vars_df['vname_id']) == len(name_list) + 1:
print('Pass: variables exist in all files')
else:
print('Fail: variables differ in between files')
#all_in_one_df = append_data(vars_df, fdatasets)
vars_df.insert(loc=len(vars_df.columns),
column='values',
value=vars_df['vname_id'])
vars_df.insert(loc=len(vars_df.columns),
column='fill_values',
value=vars_df.index)
vars_df.insert(loc=0,
column='preferred_methods_streams',
value=vars_df.index)
columns = [
'preferred_methods_streams', 'vname_id', 'units', 'var_name',
'fill_values', 'values'
]
df_time = pd.DataFrame(np.array([['', 'time', '', 'Time', '',
'time']]),
columns=columns)
vars_df = df_time.append(vars_df)
df_deployment = pd.DataFrame(np.array(
[['', 'deployment', '', 'Deployment Number', '', 'deployment']]),
columns=columns)
vars_df = df_deployment.append(vars_df)
vars_df.index = vars_df['vname_id']
all_in_one_df = pd.DataFrame()
add_to_df = pd.DataFrame()
for ii in [0, 1]: # range(len(fdatasets)):
print('\n', fdatasets[ii].split('/')[-1])
ds = xr.open_dataset(fdatasets[ii], mask_and_scale=False)
dr_ms = '-'.join((ds.collection_method, ds.stream))
# get data:
tD = ds['time'].values
print(len(tD))
dD = np.unique(ds['deployment'].values)
for vv in range(len(vars_df['vname_id'])):
if ii == 0:
vD = vars_df[vars_df['values'].values ==
vars_df['vname_id'].values[vv]]
var = ds[vD['vname_id'][0]].values
if len(np.unique(var)) == 1:
var = np.unique(var)
try:
fv = ds[vD['vname_id'][0]]._FillValue
except AttributeError:
fv = ''
vD.at[vD.index.values[0], 'values'] = var
vD.at[vD.index.values[0], 'fill_values'] = np.unique(fv)
vD.at[vD.index.values[0],
'preferred_methods_streams'] = np.unique(dr_ms)
else:
vD = add_to_df[add_to_df['vname_id'].values ==
vars_df['vname_id'].values[vv]]
var = ds[vD['vname_id'][0]].values
if len(np.unique(var)) == 1:
var = np.unique(var)
try:
fv = ds[vD['vname_id'][0]]._FillValue
except AttributeError:
fv = ''
vD.at[vD.index.values[0],
'values'] = np.append(vD['values'].values[0], var)
vD.at[vD.index.values[0], 'fill_values'] = np.append(
vD['fill_values'].values[0], fv)
vD.at[vD.index.values[0],
'preferred_methods_streams'] = np.append(
vD['preferred_methods_streams'].values[0], dr_ms)
add_to_df = add_to_df.append(vD)
all_in_one_df = add_to_df[len(add_to_df) -
len(vars_df['vname_id']):len(add_to_df)]
# vars_df.insert(loc=len(vars_df.columns), column='values', value=vars_df['vname_id'])
# #
# # columns = ['vname_id', 'units', 'var_name', 'values']
# #
# # df_time = pd.DataFrame(np.array([['time', '', 'Time', 'time']]), columns=columns)
# # vars_df = df_time.append(vars_df)
# # df_deployment = pd.DataFrame(np.array([['deployment', '', 'Deployment Number', 'deployment']]),columns=columns)
# # vars_df = df_deployment.append(vars_df)
# # vars_df.index = vars_df['vname_id']
# #
# # all_in_one_df = pd.DataFrame()
# # add_to_df = pd.DataFrame()
# #
# # for ii in [0,1]:#range(len(fdatasets)):
# # print('\n', fdatasets[ii].split('/')[-1])
# # ds = xr.open_dataset(fdatasets[ii], mask_and_scale=False)
# # dr_ms = '-'.join((ds.collection_method, ds.stream))
# # # get data:
# # tD = ds['time'].values
# # print(len(tD))
# # dD = np.unique(ds['deployment'].values)
# #
# # for vv in range(len(vars_df['vname_id'])):
# #
# # if ii == 0:
# # vD = vars_df[vars_df['values'].values == vars_df['vname_id'].values[vv]]
# # var = ds[vD['vname_id'][0]].values
# # if len(np.unique(var)) == 1:
# # var = np.unique(var)
# # vD.at[vD.index.values[0], 'values'] = var
# # else:
# # vD = add_to_df[add_to_df['vname_id'].values == vars_df['vname_id'].values[vv]]
# # var = ds[vD['vname_id'][0]].values
# # if len(np.unique(var)) == 1:
# # var = np.unique(var)
# # vD.at[vD.index.values[0], 'values'] = np.append(vD['values'].values[0], var)
# #
# # add_to_df = add_to_df.append(vD)
# #
# # all_in_one_df = add_to_df[len(add_to_df) - len(vars_df['vname_id']):len(add_to_df)]
df_roll = all_in_one_df[all_in_one_df.vname_id.str.contains('roll') ==
True]
df_pitch = all_in_one_df[all_in_one_df.vname_id.str.contains('pitch')
== True]
# select data with pitch or roll less than 20 degrees
ind = np.logical_and(df_roll['values'].values[0] < 200,
df_pitch['values'].values[0] < 200)
roll = df_roll['values'].values[0]
c_roll = roll[ind]
mean_roll, max_roll, min_roll, std_roll = np.nanmean(
c_roll), np.nanmax(c_roll), np.nanmin(c_roll), np.nanstd(c_roll)
df_U = all_in_one_df[all_in_one_df.vname_id.str.contains('eastward') ==
True]
u = df_U['values'].values[0]
uu = u[ind]
mean_u, max_u, min_u, std_u = np.nanmean(uu), np.nanmax(uu), np.nanmin(
uu), np.nanstd(uu)
df_V = all_in_one_df[all_in_one_df.vname_id.str.contains('northward')
== True]
v = df_V['values'].values[0]
vv = v[ind]
mean_v, max_v, min_v, std_v = np.nanmean(vv), np.nanmax(vv), np.nanmin(
vv), np.nanstd(vv)
df_W = all_in_one_df[all_in_one_df.vname_id.str.contains('upward') ==
True]
w = df_W['values'].values[0]
ww = w[ind]
mean_w, max_w, min_w, std_w = np.nanmean(ww), np.nanmax(ww), np.nanmin(
ww), np.nanstd(ww)
rows = []
columns = []
common_stream_name, preferred_methods_streams, deployments, [[
long_name, units
]], t0, t1, [[
fill_value, global_ranges, n_all, n_nans, n_fillvalues, n_grange
]], define_stdev, n_outliers, n_stats, mean, min, max, stdev
rows.append([
m,
list(np.unique(pms)), deployments, sv, lunits, t0, t1, fv_lst,
[g_min, g_max],
n_all, [round(ipress_min, 2),
round(ipress_max, 2)], n_excluded, n_nan, n_fv, n_grange,
sd_calc, num_outliers, n_stats, mean, vmin, vmax, sd, note
])
print('Stop')
