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-01-20 01:10:25 +01:00
parent 00da43dd5d
commit a954971e34
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/SymptomsByBatchcodesTableFactory.py
View File

@@ -6,10 +6,11 @@ class SymptomsByBatchcodesTableFactory:
@staticmethod @staticmethod
def createSymptomsByBatchcodesTable(VAERSVAX, VAERSSYMPTOMS): def createSymptomsByBatchcodesTable(VAERSVAX, VAERSSYMPTOMS):
index_columns = SymptomsByBatchcodesTableFactory._getIndexColumns(VAERSVAX) index_columns = SymptomsByBatchcodesTableFactory._getIndexColumns(VAERSVAX)
symptomsColumn = 'SYMPTOMS'
return pd.merge( return pd.merge(
SymptomsByBatchcodesTableFactory._get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns), SymptomsByBatchcodesTableFactory._get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns),
SymptomsByBatchcodesTableFactory._getSymptomsTable(VAERSSYMPTOMS), SymptomsByBatchcodesTableFactory._getSymptomsTable(VAERSSYMPTOMS, symptomsColumn),
on = 'VAERS_ID').set_index(index_columns)[['SYMPTOMS']] on = 'VAERS_ID').set_index(index_columns)[[symptomsColumn]]
@staticmethod @staticmethod
def _getIndexColumns(VAERSVAX): def _getIndexColumns(VAERSVAX):
@@ -34,7 +35,7 @@ class SymptomsByBatchcodesTableFactory:
index = VAX_LOT_LIST_Table.index) index = VAX_LOT_LIST_Table.index)
@staticmethod @staticmethod
def _getSymptomsTable(VAERSSYMPTOMS): def _getSymptomsTable(VAERSSYMPTOMS, symptomsColumn):
return pd.concat( return pd.concat(
[ [
VAERSSYMPTOMS['SYMPTOM1'], VAERSSYMPTOMS['SYMPTOM1'],
@@ -42,7 +43,7 @@ class SymptomsByBatchcodesTableFactory:
VAERSSYMPTOMS['SYMPTOM3'], VAERSSYMPTOMS['SYMPTOM3'],
VAERSSYMPTOMS['SYMPTOM4'], VAERSSYMPTOMS['SYMPTOM4'],
VAERSSYMPTOMS['SYMPTOM5'] VAERSSYMPTOMS['SYMPTOM5']
]).dropna().to_frame(name = "SYMPTOMS").reset_index() ]).dropna().to_frame(name = symptomsColumn).reset_index()
def fill(lst, desiredLen, fillValue): def fill(lst, desiredLen, fillValue):
return lst + [fillValue] * (max(desiredLen - len(lst), 0)) return lst + [fillValue] * (max(desiredLen - len(lst), 0))