Напишите функцию make_features() (англ. «создать признаки»), чтобы прибавить к таблице четыре новых календарных признака: год, месяц, день и день недели. Имена столбцов должны быть такие: 'year', 'month', 'day', 'dayofweek'.

Примените функцию к таблице и напечатайте на экране её первые пять строк (уже в прекоде).

Подсказка

Допишите код функции:

def make_features(data):
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())

Результат

             PJME_MW  year  month  day  dayofweek
Datetime                                         
2002-01-01  714857.0  2002      1    1          1
2002-01-02  822277.0  2002      1    2          2
2002-01-03  828285.0  2002      1    3          3
2002-01-04  809171.0  2002      1    4          4
2002-01-05  729723.0  2002      1    5          5
Календарь перевернули — признаки добавили.