pandas.applyで複数のカラムを引数に取る方法

例：年月日のカラムから日付カラムを追加する

importは省略

def add_date(df):
    result_df = df.copy()
    result_df['DATE'] = df[['YEAR', 'MONTH', 'DAY']].apply(lambda x: datetime.date(x[0], x[1], x[2]), axis=1)
    result_df = result_df.drop(columns=['YEAR', 'MONTH', 'DAY'])
    return result_df