refactoring

This commit is contained in:
frankknoll
2023-02-14 23:39:14 +01:00
parent 151aa9cd48
commit 11df7a0a4d

View File

@@ -9,16 +9,15 @@ class DictByBatchcodeTable2DictConverter:
@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), @staticmethod
"histogram": row['SYMPTOM_COUNT_BY_VAX_LOT'] def _addBatchcodesColumn(dictByBatchcodeTable):
}, batchcodeColumns = dictByBatchcodeTable.index.names
axis = 'columns') dictByBatchcodeTable['batchcodes'] = dictByBatchcodeTable.reset_index()[batchcodeColumns].values.tolist()
.to_list() dictByBatchcodeTable['batchcodes'] = dictByBatchcodeTable['batchcodes'].map(DictByBatchcodeTable2DictConverter._getNaNBatchcodes)
)
@staticmethod @staticmethod
def _getNaNBatchcodes(batchcodes): def _getNaNBatchcodes(batchcodes):