Agrupamento de Dados

Qual cidade vende mais? Uma única média de todas as linhas não pode dizer. Separe as linhas em grupos, um para cada cidade, e então calcule a média de cada grupo:

Uma linha faz as duas coisas:

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)

Isso imprime:

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

groupby("city") cria os grupos, ["sales"] seleciona a coluna, .mean() calcula a média de cada grupo. .sum(), .min() e .max() também funcionam.

Para classificar as pilhas, ordene-as: means.sort_values(ascending=False).index[0] é Paris, o mais vendido.