Напишите функцию make_features() (англ. «создать признаки»), чтобы прибавить к таблице четыре новых календарных признака: год, месяц, день и день недели. Имена столбцов должны быть такие: 'year', 'month', 'day', 'dayofweek'.
Примените функцию к таблице и напечатайте на экране её первые пять строк (уже в прекоде).
Подсказка
Допишите код функции:
def make_features(data):
data['year'] = data.index.year
data['month'] = data.index.month
data['day'] = # < напишите код здесь >
data['dayofweek'] = # < напишите код здесь >
import pandas as pd
import numpy as np
data = pd.read_csv('https://code.s3.yandex.net/datasets/energy_consumption.csv', index_col=[0], parse_dates=[0])
data.sort_index(inplace=True)
data = data.resample('1D').sum()
def make_features(data):
data['year'] = data.index.year
data['month'] = data.index.month
data['day'] = data.index.day
data['dayofweek'] = data.index.dayofweek
make_features(data)
print(data.head())