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

21
src/DictByBatchcodeTable2DictConverter.py
View File

@@ -6,20 +6,19 @@ class DictByBatchcodeTable2DictConverter:
"batchcode": batchcode, "batchcode": batchcode,
"histograms": DictByBatchcodeTable2DictConverter._getHistograms(dictByBatchcodeTable) "histograms": DictByBatchcodeTable2DictConverter._getHistograms(dictByBatchcodeTable)
} }
@staticmethod @staticmethod
def _getHistograms(dictByBatchcodeTable): def _getHistograms(dictByBatchcodeTable):
return ( dictByBatchcodeTable = dictByBatchcodeTable.rename(columns = { "SYMPTOM_COUNT_BY_VAX_LOT": "histogram" })
dictByBatchcodeTable DictByBatchcodeTable2DictConverter._addBatchcodesColumn(dictByBatchcodeTable)
.apply( return dictByBatchcodeTable.to_dict('records')
lambda row: {
"batchcodes": DictByBatchcodeTable2DictConverter._getNaNBatchcodes(row.name),
"histogram": row['SYMPTOM_COUNT_BY_VAX_LOT']
},
axis = 'columns')
.to_list()
)
@staticmethod
def _addBatchcodesColumn(dictByBatchcodeTable):
batchcodeColumns = dictByBatchcodeTable.index.names
dictByBatchcodeTable['batchcodes'] = dictByBatchcodeTable.reset_index()[batchcodeColumns].values.tolist()
dictByBatchcodeTable['batchcodes'] = dictByBatchcodeTable['batchcodes'].map(DictByBatchcodeTable2DictConverter._getNaNBatchcodes)
@staticmethod @staticmethod
def _getNaNBatchcodes(batchcodes): def _getNaNBatchcodes(batchcodes):
# FK-TODO: handle 'nan' everywhere correctly # FK-TODO: handle 'nan' everywhere correctly