Найти среднее значение данных одной колонки, для каждого из некоторых данных другой колонки

Дан дата фрейм с множеством колонок. Необходимо найти среднее значение данных одной колонки, для каждого из некоторых данных другой колонки без использования for. Пока имею такой код:

list_stadt = [1, 5, 8, 11]
data_looc = []
for i in [1, 5, 8, 11]:
    me_loc = data[data.A == i].location.mean()
    data_looc.append(me_loc)
print(data_looc)
data_dict = dict(zip(list_stadt, data_looc))

Результат: {1: 74370.0, 5: 65180.0, 8: 82460.0, 11: 122280.0}