francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding method HistogramTable2DictTableConverter.convertGlobalHistogramTable2DictTable(globalHistogramTable)

Browse Source
This commit is contained in:
frankknoll
2023-02-10 11:19:31 +01:00
parent 206b2a6649
commit c80d570907
2 changed files with 88 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/HistogramTable2DictTableConverter.py
View File

@@ -9,6 +9,13 @@ class HistogramTable2DictTableConverter:
.reset_index(level = 'COUNTRY')
[['SYMPTOM_COUNT_BY_VAX_LOT', 'COUNTRY']])
@staticmethod
def convertGlobalHistogramTable2DictTable(globalSymptomHistogramByBatchcodeTable):
vax_lot_columns = globalSymptomHistogramByBatchcodeTable.index.names.difference(['SYMPTOM'])
return (globalSymptomHistogramByBatchcodeTable
.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)