From f7067255291b7ac82cc22b19b7bfc2d035a15e67 Mon Sep 17 00:00:00 2001 From: frankknoll Date: Mon, 6 Feb 2023 16:05:29 +0100 Subject: [PATCH] refactoring --- src/CountryColumnAdder.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/CountryColumnAdder.py b/src/CountryColumnAdder.py index 9e703ad696c..c650ecf1662 100644 --- a/src/CountryColumnAdder.py +++ b/src/CountryColumnAdder.py @@ -4,12 +4,12 @@ class CountryColumnAdder: @staticmethod def addCountryColumn(dataFrame): - dataFrame['COUNTRY'] = CountryColumnAdder._getCountryColumn(dataFrame) + dataFrame['COUNTRY'] = CountryColumnAdder._splttype2Country(dataFrame['SPLTTYPE']) return dataFrame @staticmethod - def _getCountryColumn(dataFrame): - return (dataFrame['SPLTTYPE'] + def _splttype2Country(splttypeSeries): + return (splttypeSeries .apply( lambda splttype: CountryColumnAdder._getCountryNameOfSplttypeOrDefault(