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

formating

Browse Source
This commit is contained in:
frankknoll
2023-01-23 09:35:41 +01:00
parent ada606a61f
commit ca74a89b16
1 changed files with 22 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
src/SymptomByBatchcodeTableFactory.py
View File

@@ -7,10 +7,13 @@ class SymptomByBatchcodeTableFactory:
def createSymptomByBatchcodeTable(VAERSVAX, VAERSSYMPTOMS): def createSymptomByBatchcodeTable(VAERSVAX, VAERSSYMPTOMS):
index_columns = SymptomByBatchcodeTableFactory._getIndexColumns(VAERSVAX) index_columns = SymptomByBatchcodeTableFactory._getIndexColumns(VAERSVAX)
symptomColumn = 'SYMPTOM' symptomColumn = 'SYMPTOM'
return pd.merge( return (pd
SymptomByBatchcodeTableFactory._get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns), .merge(
SymptomByBatchcodeTableFactory._getSymptomsTable(VAERSSYMPTOMS, symptomColumn), SymptomByBatchcodeTableFactory._get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns),
on = 'VAERS_ID').set_index(index_columns)[[symptomColumn]] SymptomByBatchcodeTableFactory._getSymptomsTable(VAERSSYMPTOMS, symptomColumn),
on = 'VAERS_ID')
.set_index(index_columns)
[[symptomColumn]])
@staticmethod @staticmethod
def _getIndexColumns(VAERSVAX): def _getIndexColumns(VAERSVAX):
@@ -22,9 +25,12 @@ class SymptomByBatchcodeTableFactory:
@staticmethod @staticmethod
def _get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns): def _get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns):
return pd.concat( return (pd
[VAERSVAX, SymptomByBatchcodeTableFactory._getVaxLotsTable(VAERSVAX, index_columns)], .concat(
axis='columns').reset_index().drop_duplicates(subset = ['VAERS_ID'] + index_columns) [VAERSVAX, SymptomByBatchcodeTableFactory._getVaxLotsTable(VAERSVAX, index_columns)],
axis = 'columns')
.reset_index()
.drop_duplicates(subset = ['VAERS_ID'] + index_columns))
@staticmethod @staticmethod
def _getVaxLotsTable(VAERSVAX, index_columns): def _getVaxLotsTable(VAERSVAX, index_columns):
@@ -42,14 +48,15 @@ class SymptomByBatchcodeTableFactory:
@staticmethod @staticmethod
def _getSymptomsTable(VAERSSYMPTOMS, symptomColumn): def _getSymptomsTable(VAERSSYMPTOMS, symptomColumn):
return (pd.concat( return (pd
[ .concat(
VAERSSYMPTOMS['SYMPTOM1'], [
VAERSSYMPTOMS['SYMPTOM2'], VAERSSYMPTOMS['SYMPTOM1'],
VAERSSYMPTOMS['SYMPTOM3'], VAERSSYMPTOMS['SYMPTOM2'],
VAERSSYMPTOMS['SYMPTOM4'], VAERSSYMPTOMS['SYMPTOM3'],
VAERSSYMPTOMS['SYMPTOM5'] VAERSSYMPTOMS['SYMPTOM4'],
]) VAERSSYMPTOMS['SYMPTOM5']
])
.dropna() .dropna()
.to_frame(name = symptomColumn) .to_frame(name = symptomColumn)
.reset_index()) .reset_index())