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

refining HistogramTable2DictTableConverterTest

Browse Source
This commit is contained in:
frankknoll
2023-02-08 16:33:04 +01:00
parent 197321b2da
commit 16e6c2a729
2 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/HistogramTable2DictTableConverter.py
View File

@@ -3,11 +3,11 @@ class HistogramTable2DictTableConverter:
@staticmethod
def convertHistogramTable2DictTable(symptomHistogramByBatchcodeTable):
vax_lot_columns = symptomHistogramByBatchcodeTable.index.names.difference(['SYMPTOM'])
return (
symptomHistogramByBatchcodeTable
.groupby(vax_lot_columns)
return (symptomHistogramByBatchcodeTable
.groupby(vax_lot_columns + ['COUNTRY'])
.agg(lambda histogram_with_vax_lots: HistogramTable2DictTableConverter._histogram_to_json(histogram_with_vax_lots, vax_lot_columns))
)
.reset_index(level = 'COUNTRY')
[['SYMPTOM_COUNT_BY_VAX_LOT', 'COUNTRY']])
@staticmethod
def _histogram_to_json(histogram_with_vax_lots, vax_lot_columns):