Add "median" option to group by and pivot plugins

CountESS-Project · Sep 19, 2024 · 689c12f · 689c12f
1 parent 44dd79f
commit 689c12f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/countess/plugins/group_by.py b/countess/plugins/group_by.py
@@ -19,6 +19,7 @@ class ColumnMultiParam(TabularMultiParam):
     max = BooleanParam("Max")
     sum = BooleanParam("Sum")
     mean = BooleanParam("Mean")
+    median = BooleanParam("Median")
 
 
 class GroupByPlugin(PandasConcatProcessPlugin):

diff --git a/countess/plugins/pivot.py b/countess/plugins/pivot.py
@@ -29,7 +29,7 @@ class PivotPlugin(PandasProcessPlugin):
     link = "https://countess-project.github.io/CountESS/included-plugins/#pivot-tool"
 
     columns = PerColumnArrayParam("Columns", ChoiceParam("Role", "Drop", choices=["Index", "Pivot", "Expand", "Drop"]))
-    aggfunc = ChoiceParam("Aggregation Function", "sum", choices=["sum", "mean", "min", "max"])
+    aggfunc = ChoiceParam("Aggregation Function", "sum", choices=["sum", "mean", "median", "min", "max"])
 
     input_columns: Dict[str, np.dtype] = {}