snapshot...

modules/aeolus/aeolus_amv.py

@@ -1062,12 +1062,6 @@ def compare_amvs_bestfit_all(all_list, bin_size=200):
     #         amvs_list.append(tup[0])
     #         bfs_list.append(tup[1])
 
-    amvs = np.concatenate(amvs_list)
-    bfs = np.concatenate(bfs_list)
-    print(amvs.shape, bfs.shape)
-    prd = np.concatenate(prd_list)
-
-    print(amvs.shape, bfs.shape, prd.shape)
 
     # for tup in all_list:
     #     amvs_list.append(tup[0])
@@ -1075,7 +1069,26 @@ def compare_amvs_bestfit_all(all_list, bin_size=200):
     # amvs = np.concatenate(amvs_list)
     # bfs = np.concatenate(bfs_list)
 
+    amvs = np.concatenate(amvs_list)
+    bfs = np.concatenate(bfs_list)
+    prd = np.concatenate(prd_list)
+
+    thin = np.logical_and(prd[:, 2] > 0, prd[:, 2] < 1)
+    thick = np.logical_and(prd[:, 2] >= 1, prd[:, 2] < 18)
+    deep = np.logical_and(prd[:, 2] >= 18, prd[:, 2] <= 20)
+
+    amvs_thin = amvs[thin, :]
+    amvs_thick = amvs[thick, :]
+    amvs_deep = amvs[deep, :]
+
+    bfs_thin = bfs[thin, :]
+    bfs_thick = bfs[thick, :]
+    bfs_deep = bfs[deep, :]
+
     bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs, bfs, bin_size=bin_size)
+    bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_thin, bfs_thin, bin_size=bin_size)
+    bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_thick, bfs_thick, bin_size=bin_size)
+    bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_deep, bfs_deep, bin_size=bin_size)
 
 
 def compare_amvs_bestfit(amvs, bfs, bin_size=200):