Порядковая кодировка столбца в PolarsPython

Программы на Python
Anonymous
Порядковая кодировка столбца в Polars

Сообщение Anonymous »

Я хотел бы выполнить порядковую кодировку столбца. В Pandas есть красивый и удобный метод pd.factorize(), однако я хотел бы добиться того же в полярах.
import polars as pl

df = pl.DataFrame({"a": [5, 8, 10], "b": ["hi", "hello", "hi"]})

┌─────┬───────┐
│ a ┆ b │
│ --- ┆ --- │
│ i64 ┆ str │
╞═════╪═══════╡
│ 5 ┆ hi │
│ 8 ┆ hello │
│ 10 ┆ hi │
└─────┴───────┘

желаемый результат:
┌─────┬─────┐
│ a ┆ b │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 0 ┆ 0 │
│ 1 ┆ 1 │
│ 2 ┆ 0 │
└─────┴─────┘


Подробнее здесь: https://stackoverflow.com/questions/746 ... -in-polars

Вернуться в «Python»