polars.Series.cut

Series.cut(
    breaks: Sequence[float],
    *,
    labels: Sequence[str_] | None = None,
    left_closed: bool = False,
    include_breaks: bool = False,
) → Series

Разбивает непрерывные значения на дискретные категории.

Предупреждение

Эта функция считается нестабильной. Она может быть изменена в любой момент без необходимости считать такое изменение нарушающим совместимость.

Параметры:

breaks

Список уникальных точек разбиения.

labels

Названия категорий. Количество меток должно быть равно количеству точек разбиения плюс один.

left_closed

Задаёт интервалы с замкнутой левой границей вместо замкнутой правой.

include_breaks

Добавляет столбец с правой границей интервала, в который попадает каждое наблюдение. Это изменит тип данных результата с Enum на Struct.

Возвращает:

Series

Series типа данных Enum, если для include_breaks задано значение False (по умолчанию); в противном случае возвращает Series типа данных Struct.

См. также

qcut

Примеры

Разделение столбца на три категории.

>>> s = pl.Series("foo", [-2, -1, 0, 1, 2])
>>> s.cut([-1, 1], labels=["a", "b", "c"])
shape: (5,)
Series: 'foo' [enum]
[
    "a"
    "a"
    "b"
    "b"
    "c"
]

Создание DataFrame с точкой разбиения и категорией для каждого значения.

>>> cut = s.cut([-1, 1], include_breaks=True).alias("cut")
>>> s.to_frame().with_columns(cut).unnest("cut")
shape: (5, 3)
┌─────┬────────────┬────────────┐
│ foo ┆ breakpoint ┆ category   │
│ --- ┆ ---        ┆ ---        │
│ i64 ┆ f64        ┆ enum       │
╞═════╪════════════╪════════════╡
│ -2  ┆ -1.0       ┆ (-inf, -1] │
│ -1  ┆ -1.0       ┆ (-inf, -1] │
│ 0   ┆ 1.0        ┆ (-1, 1]    │
│ 1   ┆ 1.0        ┆ (-1, 1]    │
│ 2   ┆ inf        ┆ (1, inf]   │
└─────┴────────────┴────────────┘