Agrupamento de Números em Faixas

Uma companhia aérea cobra pela bagagem de acordo com sua faixa de peso, não por grama. O agrupamento em faixas transforma números em intervalos, chamados de faixas, quando apenas o intervalo importa:

pd.cut recebe os valores, os limites e um rótulo por intervalo:

bags = pd.DataFrame({"kg": [5.5, 23.0, 18.2, 30.5, 7.0, 12.4]})
bags["class"] = pd.cut(bags["kg"], bins=[0, 7, 23, 32],
                       labels=["cabin", "standard", "heavy"])
print(bags["class"].value_counts().sort_index())

Isso imprime:

class
cabin       2
standard    3
heavy       1
Name: count, dtype: int64

Cada intervalo inclui sua borda direita, então 7,0 kg ainda é bagagem de cabine, e um valor fora das bordas se torna NaN.

.sort_index() lista os intervalos em sua própria ordem, do mais claro ao mais escuro.