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))