refactoring

2023-02-06 16:05:29 +01:00
parent 971f01935a
commit f706725529
1 changed files with 3 additions and 3 deletions
--- 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(