francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
frankknoll
2023-02-06 15:58:32 +01:00
parent d56b1b8a9d
commit 971f01935a
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/CountryColumnAdder.py
View File

@@ -4,17 +4,18 @@ class CountryColumnAdder:
@staticmethod @staticmethod
def addCountryColumn(dataFrame): def addCountryColumn(dataFrame):
dataFrame['COUNTRY'] = CountryColumnAdder.getCountryColumn(dataFrame) dataFrame['COUNTRY'] = CountryColumnAdder._getCountryColumn(dataFrame)
return dataFrame.astype({'COUNTRY': "string"}) return dataFrame
@staticmethod @staticmethod
def getCountryColumn(dataFrame): def _getCountryColumn(dataFrame):
return dataFrame.apply( return (dataFrame['SPLTTYPE']
lambda row: .apply(
lambda splttype:
CountryColumnAdder._getCountryNameOfSplttypeOrDefault( CountryColumnAdder._getCountryNameOfSplttypeOrDefault(
splttype = row['SPLTTYPE'], splttype = splttype,
default = 'Unknown Country'), default = 'Unknown Country'))
axis = 'columns') .astype("string"))
@staticmethod @staticmethod
def _getCountryNameOfSplttypeOrDefault(splttype, default): def _getCountryNameOfSplttypeOrDefault(splttype, default):