Как посчитать weekday avarage?

Датасет https://drive.google.com/file/d/1HVPWEi6AknmOVBPWheizOG9oI225C3-C/view?usp=sharing По примеру статьи https://habr.com/ru/company/ods/blog/327242/

import sys
import warnings
warnings.filterwarnings('ignore')
from tqdm import tqdm

import pandas as pd
import numpy as np
from sklearn.metrics import mean_absolute_error, mean_squared_error

import statsmodels.formula.api as smf
import statsmodels.tsa.api as smt
import statsmodels.api as sm
import scipy.stats as scs
from scipy.optimize import minimize

import matplotlib.pyplot as plt
def code_mean(data, cat_feature, real_feature):
    """
    Возвращает словарь, где ключами являются уникальные категории признака cat_feature, 
    а значениями - средние по real_feature
    """
return dict(data.groupby(cat_feature)[real_feature].mean())

dataset = pd.read_csv('/content/drive/My Drive/coursera/hour_online.csv')
data = dataset.copy()
data.columns = ["Time","y"]
data['Time'] = pd.to_datetime(data['Time'])
data["hour"] = data['Time'].dt.hour
data["weekday"] = data['Time'].dt.weekday
data['is_weekend'] = data.weekday.isin([5,6])*1
data.head()
lag_start=5 
lag_end=20
test_size=0.15

test_index = int(len(data.y)*(1-test_size))
for i in range(lag_start, lag_end):
    data["lag_{}".format(i)] = data.y.shift(i)
data['weekday_average'] = map(code_mean(data[:test_index], 'weekday', "y").get, data.weekday)
data["hour_average"] = map(code_mean(data[:test_index], 'hour', "y").get, data.hour)
data.drop(["hour", "weekday"], axis=1, inplace=True)

data = data.dropna()
data = data.reset_index(drop=True)
X_train = data.loc[:test_index].drop(["y"], axis=1)
y_train = data.loc[:test_index]["y"]
X_test = data.loc[test_index:].drop(["y"], axis=1)
y_test = data.loc[test_index:]["y"]
from sklearn.linear_model import LinearRegression
lr = LinearRegression()
lr.fit(X_train, y_train)
X_train = X_train.fillna(0)
X_test = X_test.fillna(0)

TypeError: invalid type promotion

data['weekday_average'] = map(code_mean(data[:test_index], 'weekday', "y").get, data.weekday)
data["hour_average"] = map(code_mean(data[:test_index], 'hour', "y").get, data.hour) 

Возвращают объект map, LinearRegression с таким работать не умеет. Как сделать отдельные колонки weekday_average_1 (среднее значение для дня 1), weekday_average_12(среднее значение для дня 2), и т. д.?