Files
HowBadIsMyBatch/src/CountryColumnAdder.py
frankknoll 3f0b650519 refactoring
2023-02-07 11:37:43 +01:00

16 lines
540 B
Python

import pandas as pd
from Splttype2CountryConverter import Splttype2CountryConverter
class CountryColumnAdder:
def __init__(self, dataFrame_SPLTTYPE_By_VAERS_ID):
self.dataFrame_COUNTRY_By_VAERS_ID = Splttype2CountryConverter.convertSplttype2Country(dataFrame_SPLTTYPE_By_VAERS_ID)
def addCountryColumn(self, dataFrame):
return pd.merge(
dataFrame,
self.dataFrame_COUNTRY_By_VAERS_ID,
how = 'left',
left_index = True,
right_index = True)