From ada606a61fb8a1492f20d81f0d2ad979ec297772 Mon Sep 17 00:00:00 2001 From: frankknoll Date: Mon, 23 Jan 2023 09:30:04 +0100 Subject: [PATCH] formating --- src/SymptomByBatchcodeTableFactory.py | 19 +++++++++++-------- ...SymptomHistogramByBatchcodeTableFactory.py | 2 +- 2 files changed, 12 insertions(+), 9 deletions(-) diff --git a/src/SymptomByBatchcodeTableFactory.py b/src/SymptomByBatchcodeTableFactory.py index e6a61e47555..d44cbc14d4d 100644 --- a/src/SymptomByBatchcodeTableFactory.py +++ b/src/SymptomByBatchcodeTableFactory.py @@ -42,14 +42,17 @@ class SymptomByBatchcodeTableFactory: @staticmethod def _getSymptomsTable(VAERSSYMPTOMS, symptomColumn): - return pd.concat( - [ - VAERSSYMPTOMS['SYMPTOM1'], - VAERSSYMPTOMS['SYMPTOM2'], - VAERSSYMPTOMS['SYMPTOM3'], - VAERSSYMPTOMS['SYMPTOM4'], - VAERSSYMPTOMS['SYMPTOM5'] - ]).dropna().to_frame(name = symptomColumn).reset_index() + return (pd.concat( + [ + VAERSSYMPTOMS['SYMPTOM1'], + VAERSSYMPTOMS['SYMPTOM2'], + VAERSSYMPTOMS['SYMPTOM3'], + VAERSSYMPTOMS['SYMPTOM4'], + VAERSSYMPTOMS['SYMPTOM5'] + ]) + .dropna() + .to_frame(name = symptomColumn) + .reset_index()) def fillLsts(lsts, desiredLen, fillValue): return [fillLst(lst, desiredLen, fillValue) for lst in lsts] diff --git a/src/SymptomHistogramByBatchcodeTableFactory.py b/src/SymptomHistogramByBatchcodeTableFactory.py index 8df21f09ffd..de563b08d7f 100644 --- a/src/SymptomHistogramByBatchcodeTableFactory.py +++ b/src/SymptomHistogramByBatchcodeTableFactory.py @@ -5,4 +5,4 @@ class SymptomHistogramByBatchcodeTableFactory: return (symptomByBatchcodeTable .groupby(symptomByBatchcodeTable.index.names) ['SYMPTOM'].value_counts() - .to_frame('SYMPTOM_COUNT_BY_VAX_LOT')) + .to_frame(name = 'SYMPTOM_COUNT_BY_VAX_LOT'))