Add results columns for max_ebs_hits and mean_ebs_hits (#55)

matsengrp · Feb 17, 2023 · 0f5e0be · 0f5e0be
1 parent cf85e52
commit 0f5e0be
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/templates/aggregate_organisms.py b/templates/aggregate_organisms.py
@@ -326,6 +326,7 @@ def group_sample_organisms(self, df:pd.DataFrame, sample:str, organism:str) -> p
                 for label, d in [
                     ("all", df),
                     ("public", df.query("public")),
+                    ("hits", df.query("hit == 'TRUE'")),
                 ]
                 if d.shape[0] > 0
                 for k, v in [
@@ -334,6 +335,7 @@ def group_sample_organisms(self, df:pd.DataFrame, sample:str, organism:str) -> p
                     (f"max_ebs_{label}", d["EBS"].max()),
                     (f"mean_ebs_{label}", d["EBS"].mean())
                 ]
+                if k not in ["n_hits_hits", "n_discordant_hits"]
             }
         }])