From c3ad15a24ce4374f3b8ab868fe45e0810cd1719f Mon Sep 17 00:00:00 2001 From: frankknoll Date: Mon, 6 Feb 2023 12:31:20 +0100 Subject: [PATCH] refactoring --- src/SymptomByBatchcodeTableFactory.py | 7 +------ src/Utils.py | 6 ++++++ 2 files changed, 7 insertions(+), 6 deletions(-) create mode 100644 src/Utils.py diff --git a/src/SymptomByBatchcodeTableFactory.py b/src/SymptomByBatchcodeTableFactory.py index 9670b5369b1..454d4bea347 100644 --- a/src/SymptomByBatchcodeTableFactory.py +++ b/src/SymptomByBatchcodeTableFactory.py @@ -1,5 +1,6 @@ import pandas as pd import numpy as np +from Utils import fillLsts class SymptomByBatchcodeTableFactory: @@ -73,9 +74,3 @@ class SymptomByBatchcodeTableFactory: .dropna() .to_frame(name = symptomColumn) .reset_index()) - -def fillLsts(lsts, desiredLen, fillValue): - return [fillLst(lst, desiredLen, fillValue) for lst in lsts] - -def fillLst(lst, desiredLen, fillValue): - return lst + [fillValue] * (max(desiredLen - len(lst), 0)) \ No newline at end of file diff --git a/src/Utils.py b/src/Utils.py new file mode 100644 index 00000000000..51ac94b8ab7 --- /dev/null +++ b/src/Utils.py @@ -0,0 +1,6 @@ +def fillLsts(lsts, desiredLen, fillValue): + return [fillLst(lst, desiredLen, fillValue) for lst in lsts] + + +def fillLst(lst, desiredLen, fillValue): + return lst + [fillValue] * (max(desiredLen - len(lst), 0))