Spaces:
Sleeping
Sleeping
File size: 580 Bytes
fa10c3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
def prepare_data(
dataframe,
add_datetime_features=True
):
print('Building features...')
if add_datetime_features:
dataframe = datetime_features(dataframe)
return dataframe
def datetime_features(dataframe, date='date', suffix=''):
dataframe[f'{suffix}_month'] = dataframe[date].dt.month
dataframe[f'{suffix}_year'] = dataframe[date].dt.year
dataframe[f'{suffix}_quarter'] = dataframe[date].dt.quarter
dataframe[f'{suffix}_weekofyear'] = dataframe[date].dt.isocalendar().week
return dataframe |