Closed passi3 closed 1 month ago
from src.pre_process.interest.diff_interestrate import DiffInterestRateAdder
BASE_PATH = os.path.join("..", "..", "data", "raw")
train_data = pd.read_csv(os.path.join(BASE_PATH, 'train.csv'))
interest_data = pd.read_csv(os.path.join(BASE_PATH, 'interestRate.csv'))
adder = DiffInterestRateAdder(train_data, interest_data)
adder.get_data()
LGTM👍👍👍
두 개의 데이터프레임을 입력 인자로 반환받아 기준이 될 datetime 형식의 칼럼이 있는지를 확인합니다. datetime 형식이 없을 경우, 칼럼명에 year_month가 있는지를 확인하고 해당 칼럼명을 반환합니다. 또한, 금리 정보를 담은 interest_rate 칼럼이 포함된 데이터프레임에서 datetime 형식인 칼럼 기준으로 정렬&병합합니다.