From 4401a68890e9bc813086e1c5c849ea84795e2352 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 6 Jan 2021 10:50:34 -0600
Subject: [PATCH] snapshot...
---
modules/aeolus/aeolus_amv.py | 65 ++++++++++++++++++++++++++++++++++--
1 file changed, 62 insertions(+), 3 deletions(-)
diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py
index 0f85cd51..b1f69cbe 100644
--- a/modules/aeolus/aeolus_amv.py
+++ b/modules/aeolus/aeolus_amv.py
@@ -1017,6 +1017,65 @@ def compare_amvs_bestfit_driver(amvs, bfs, prd, bfs_gfs, rbm, bin_size=200):
return bin_ranges, values_prs, values_spd, values_dir
+def compare_amvs_bestfit_driver2(amvs, bfs, prd, bfs_gfs, rbm, bin_size=200):
+
+ good = np.logical_and(prd[:, 0] != 5, prd[:, 0] != 0)
+ ice = np.logical_and(good, prd[:, 0] == 4)
+ not_ice = np.logical_and(good, prd[:, 0] != 4)
+
+ amvs_ice = amvs[ice, :]
+ amvs_not_ice = amvs[not_ice, :]
+
+ bfs_ice = bfs[ice, :]
+ bfs_not_ice = bfs[not_ice, :]
+
+ # bfs_gfs_ice = bfs_gfs[ice, :]
+ # bfs_gfs_not_ice = bfs_gfs[not_ice, :]
+ #
+ # rbm_ice = rbm[ice, :]
+ # rbm_not_ice = rbm[thick, :]
+
+ # values_pres = []
+ # bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_thin, bfs_thin, bin_size=bin_size)
+ # values_pres.append(bin_pres)
+ #
+ # bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_thick, bfs_thick, bin_size=bin_size)
+ # values_pres.append(bin_pres)
+ #
+ # bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs, bfs, bin_size=bin_size)
+ # values_pres.append(bin_pres)
+ #
+ # values_spd = []
+ # bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_thin, bfs_gfs_thin, bin_size=bin_size)
+ # values_spd.append(bin_spd)
+ #
+ # bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_thick, bfs_gfs_thick, bin_size=bin_size)
+ # values_spd.append(bin_spd)
+ #
+ # bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs, bfs_gfs, bin_size=bin_size)
+ # values_spd.append(bin_spd)
+
+ values_prs = []
+ values_spd = []
+ values_dir = []
+ bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_ice, bfs_ice, bin_size=bin_size)
+ values_prs.append(bin_pres)
+ values_spd.append(bin_spd)
+ values_dir.append(bin_dir)
+
+ bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_not_ice, bfs_not_ice, bin_size=bin_size)
+ values_prs.append(bin_pres)
+ values_spd.append(bin_spd)
+ values_dir.append(bin_dir)
+
+ bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs, bfs, bin_size=bin_size)
+ values_prs.append(bin_pres)
+ values_spd.append(bin_spd)
+ values_dir.append(bin_dir)
+
+ return bin_ranges, values_prs, values_spd, values_dir
+
+
def compare_amvs_bestfit(amvs, bfs, bin_size=200):
good_amvs = amvs
@@ -1186,15 +1245,15 @@ def make_plot(bin_ranges, bin_values):
std.append(np.std(bin_vals[i]))
- do_plot(x_values, [std_r, std_g, std], ['THIN', 'THICK', 'ALL'], ['blue', 'red', 'black'], title='ACHA - RAOB Pres Match', x_axis_label='STD (hPa)', y_axis_label='hPa', invert=True, flip=True)
+ #do_plot(x_values, [std_r, std_g, std], ['THIN', 'THICK', 'ALL'], ['blue', 'red', 'black'], title='ACHA - RAOB Pres Match', x_axis_label='STD (hPa)', y_axis_label='hPa', invert=True, flip=True)
#do_plot(x_values, [std_r, std_g, std], ['THIN', 'THICK', 'ALL'], ['blue', 'red', 'black'], title='ACHA - RAOB Pres Match', x_axis_label='STD (deg)', y_axis_label='hPa', invert=True, flip=True)
#do_plot(x_values, [mad_r, mad_g, mad], ['THIN', 'THICK', 'ALL'], ['blue', 'red', 'black'], title='ACHA - RAOB Pres Match', x_axis_label='MAD (deg)', y_axis_label='hPa', invert=True, flip=True)
- #do_plot(x_values, [mad_r, mad_g, mad], ['THIN', 'THICK', 'ALL'], ['blue', 'red', 'black'], title='ACHA - BestFit Full Domain (GFS)', x_axis_label='MAD', y_axis_label='hPa', invert=True, flip=True)
+ #do_plot(x_values, [mad_r, mad_g, mad], ['ICE', 'NOT ICE', 'ALL'], ['blue', 'red', 'black'], title='ACHA - BestFit Full Domain (GFS)', x_axis_label='MAD', y_axis_label='hPa', invert=True, flip=True)
#do_plot(x_values, [pres_mad_r, pres_mad_g], ['RAOB', 'GFS'], ['blue', 'red'], title='ACHA - BestFit', x_axis_label='MAD', y_axis_label='hPa', invert=True, flip=True)
#do_plot(x_values, [spd_mad_r, spd_mad_g], ['RAOB', 'GFS'], ['blue', 'red'], title='ACHA - BestFit', x_axis_label='MAE (m/s)', y_axis_label='hPa', invert=True, flip=True)
#do_plot(x_values, [pres_bias_r, pres_bias_g], ['RAOB', 'GFS'], ['blue', 'red'], title='ACHA - BestFit', x_axis_label='BIAS', y_axis_label='hPa', invert=True, flip=True)
#do_plot(x_values, [spd_bias_r, spd_bias_g], ['RAOB', 'GFS'], ['blue', 'red'], title='ACHA - BestFit', x_axis_label='BIAS (m/s)', y_axis_label='hPa', invert=True, flip=True)
- #do_plot(x_values, [num_vals_r, num_vals_g, num_vals], ['THIN: '+str(num_r), 'THICK: '+str(num_g), 'ALL: '+str(num)], ['blue', 'red', 'black'], title='ACHA - RAOB PRES MATCH', x_axis_label='log(Count)', y_axis_label='hPa', invert=True, flip=True)
+ do_plot(x_values, [num_vals_r, num_vals_g, num_vals], ['ICE: '+str(num_r), 'NOT ICE: '+str(num_g), 'ALL: '+str(num)], ['blue', 'red', 'black'], title='ACHA - BestFit Full Domain (GFS)', x_axis_label='log(Count)', y_axis_label='hPa', invert=True, flip=True)
def make_plot2(bin_pres_r, bin_ranges):
--
GitLab