Разбиение чисел на интервалы

Авиакомпания взимает плату за багаж в зависимости от весовой категории, а не от количества граммов. Разбиение на интервалы превращает числа в диапазоны, называемые интервалами, когда важно только то, в какой диапазон попадает значение:

pd.cut принимает значения, границы и одну метку для каждого интервала:

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

Это выводит:

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

Каждый интервал включает свою правую границу, поэтому 7.0 kg всё ещё относится к ручной клади, а значение за пределами границ становится NaN.

.sort_index() перечисляет интервалы в их собственном порядке, начиная с самого лёгкого.