Группировка данных

В каком городе продают больше? Одно среднее значение по всем строкам этого не покажет. Раздели строки на группы — по одной для каждого города, — а затем вычисли среднее значение для каждой группы:

Одна строка делает и то и другое:

shops = pd.DataFrame({
    "city": ["Lyon", "Paris", "Lyon", "Nice", "Paris", "Lyon"],
    "sales": [20, 35, 26, 18, 41, 23],
})
means = shops.groupby("city")["sales"].mean()
print(means)

Это выводит:

city
Lyon     23.0
Nice     18.0
Paris    38.0
Name: sales, dtype: float64

groupby("city") формирует группы, ["sales"] выбирает столбец, .mean() вычисляет среднее значение для каждой группы. Также работают .sum(), .min() и .max().

Чтобы ранжировать стопки, отсортируй их: means.sort_values(ascending=False).index[0] — это Paris, лидер продаж.