polars.Series.to_dummies

Series.to_dummies(
    *,
    separator: str_ = '_',
    drop_first: bool = False,
    drop_nulls: bool = False,
) → DataFrame

Получить фиктивные/индикаторные переменные.

Параметры:

separator

Разделитель, используемый при создании имён столбцов.

drop_first

Удалить первую категорию из кодируемой переменной.

drop_nulls

Если в серии есть значения None, столбец null не создаётся. Значения null во входных данных представлены нулевыми векторами.

Примеры

>>> s = pl.Series("a", [1, 2, 3])
>>> s.to_dummies()
shape: (3, 3)
┌─────┬─────┬─────┐
│ a_1 ┆ a_2 ┆ a_3 │
│ --- ┆ --- ┆ --- │
│ u8  ┆ u8  ┆ u8  │
╞═════╪═════╪═════╡
│ 1   ┆ 0   ┆ 0   │
│ 0   ┆ 1   ┆ 0   │
│ 0   ┆ 0   ┆ 1   │
└─────┴─────┴─────┘

>>> s.to_dummies(drop_first=True)
shape: (3, 2)
┌─────┬─────┐
│ a_2 ┆ a_3 │
│ --- ┆ --- │
│ u8  ┆ u8  │
╞═════╪═════╡
│ 0   ┆ 0   │
│ 1   ┆ 0   │
│ 0   ┆ 1   │
└─────┴─────┘

>>> s = pl.Series("a", [1, 2, None, 3])
>>> s.to_dummies(drop_nulls=True, drop_first=True)
shape: (4, 2)
┌─────┬─────┐
│ a_2 ┆ a_3 │
│ --- ┆ --- │
│ u8  ┆ u8  │
╞═════╪═════╡
│ 0   ┆ 0   │
│ 1   ┆ 0   │
│ 0   ┆ 0   │
│ 0   ┆ 1   │
└─────┴─────┘