- starting DictByBatchcodeTable2JsonConverterTest

- refactoring
2023-01-26 22:16:57 +01:00
parent 4b38b0a6b7
commit a87af43c58
4 changed files with 97 additions and 11 deletions
--- a/src/HistogramTable2DictTableConverter.py
+++ b/src/HistogramTable2DictTableConverter.py
@@ -0,0 +1,15 @@
+class HistogramTable2DictTableConverter:
+
+    @staticmethod
+    def convertHistogramTable2DictTable(symptomHistogramByBatchcodeTable):
+        vax_lot_columns = symptomHistogramByBatchcodeTable.index.names.difference(['SYMPTOM'])
+        return (
+            symptomHistogramByBatchcodeTable
+                    .groupby(vax_lot_columns)
+                    .agg(lambda histogram_with_vax_lots: HistogramTable2DictTableConverter._histogram_to_json(histogram_with_vax_lots, vax_lot_columns))
+        )
+
+    @staticmethod
+    def _histogram_to_json(histogram_with_vax_lots, vax_lot_columns):
+        histogram = histogram_with_vax_lots.reset_index(level = vax_lot_columns, drop=True)
+        return histogram.to_dict()