#соберем сводную таблицу методом .groupby

agg_data = all_in_one.groupby(['user_id','month']).agg({'distance':'sum','duration':['sum','count']})

#наведем красоту в названиях колонок и идексах

agg_data.columns = ['distance_sum', 'duration_sum', 'rides_count']

agg_data = agg_data.reset_index()

#объявляем функцию расчета выручки в месяц

def income_func (row):

    if row['user_id'] in riders_ultra['user_id']:

        return 6*row['duration_sum'].round()+199

    if row['user_id'] in riders_free['user_id']:

        return 8*row['duration_sum']+ 50*row['rides_count']

#добавим столбец со стоимостью поездок, применив функцию к каждой строке agg_data

agg_data['income'] = agg_data.apply(income_func, axis=1)      

print(agg_data.head(15))

print(agg_data.tail(15))