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

refactoring

Browse Source
This commit is contained in:
frankknoll
2023-02-14 23:39:14 +01:00
parent 151aa9cd48
commit 11df7a0a4d
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/DictByBatchcodeTable2DictConverter.py
View File

@@ -9,16 +9,15 @@ class DictByBatchcodeTable2DictConverter:
@staticmethod
def _getHistograms(dictByBatchcodeTable):
return (
dictByBatchcodeTable
.apply(
lambda row: {
"batchcodes": DictByBatchcodeTable2DictConverter._getNaNBatchcodes(row.name),
"histogram": row['SYMPTOM_COUNT_BY_VAX_LOT']
},
axis = 'columns')
.to_list()
)
dictByBatchcodeTable = dictByBatchcodeTable.rename(columns = { "SYMPTOM_COUNT_BY_VAX_LOT": "histogram" })
DictByBatchcodeTable2DictConverter._addBatchcodesColumn(dictByBatchcodeTable)
return dictByBatchcodeTable.to_dict('records')
@staticmethod
def _addBatchcodesColumn(dictByBatchcodeTable):
batchcodeColumns = dictByBatchcodeTable.index.names
dictByBatchcodeTable['batchcodes'] = dictByBatchcodeTable.reset_index()[batchcodeColumns].values.tolist()
dictByBatchcodeTable['batchcodes'] = dictByBatchcodeTable['batchcodes'].map(DictByBatchcodeTable2DictConverter._getNaNBatchcodes)
@staticmethod
def _getNaNBatchcodes(batchcodes):