Don't process omitted profiles

3 years ago · 5e2cfccb34
parent 94d150d957
commit 5e2cfccb34
1 changed files with 6 additions and 0 deletions
--- a/probabilistic-analysis/probabilistic_assessment.py
+++ b/probabilistic-analysis/probabilistic_assessment.py
@ -438,6 +438,12 @@ def process(beach_name, beach_scenario, n_runs, start_year, end_year,

        # Loop through profiles
        dff = df_in[df_in['beach'] == beach_name]
+
+        # Remove omitted profiles
+        dff = pd.merge(
+            pd.DataFrame(omit), dff, how='outer',
+            indicator='source').query('source!="both"').drop(columns='source')
+
        pbar_profile = tqdm(dff.iterrows(), total=dff.shape[0], leave=False)
        for i, prof in pbar_profile: