Regrouper les nombres en intervalles

Une compagnie aérienne facture un bagage selon sa catégorie de poids, et non au gramme près. Le regroupement en intervalles transforme les nombres en plages, appelées intervalles, lorsque seule la plage compte :

pd.cut prend les valeurs, les limites et une étiquette par intervalle :

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())

Cela affiche :

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

Chaque intervalle inclut son bord droit, donc 7.0 kg est toujours considéré comme bagage cabine, et une valeur située en dehors des limites devient NaN.

.sort_index() liste les catégories dans leur propre ordre, de la plus claire à la plus foncée.