Возьмите уникальные значения по горизонтали в кадре данных Polars, чтобы создать новый строковый столбец [дубликат]Python

Программы на Python
Anonymous
 Возьмите уникальные значения по горизонтали в кадре данных Polars, чтобы создать новый строковый столбец [дубликат]

Сообщение Anonymous »

У меня есть этот фрейм данных:

Код: Выделить всё

import polars as pl

df = pl.from_repr("""shape: (4, 3)
┌──────┬──────┐
│ ccy1 ┆ ccy2 │
│ ---  ┆ ---  │
│ str  ┆ str  │
╞══════╪══════╡
│ USD  ┆ USD  │
│ EUR  ┆ USD  │
│ EUR  ┆ EUR  │
│ USD  ┆ JPY  │
└──────┴──────┘
""")
Я хочу создать третий столбец, который будет содержать объединенный список уникальных значений в столбцах ccy1 и ccy2. Что-то вроде

Код: Выделить всё

df.with_columns(pl.struct('ccy1', 'ccy2')
.map_elements(lambda x: ','.join(sorted(set(x.values()))),
return_dtype=pl.String)
.alias('ccys'))
что дает

Код: Выделить всё

shape: (4, 3)
┌──────┬──────┬─────────┐
│ ccy1 ┆ ccy2 ┆ ccys    │
│ ---  ┆ ---  ┆ ---     │
│ str  ┆ str  ┆ str     │
╞══════╪══════╪═════════╡
│ USD  ┆ USD  ┆ USD     │
│ EUR  ┆ USD  ┆ EUR,USD │
│ EUR  ┆ EUR  ┆ EUR     │
│ USD  ┆ JPY  ┆ JPY,USD │
└──────┴──────┴─────────┘
Но я ищу более естественный полярный способ (т. е. векторизованный). Любая помощь приветствуется.

Подробнее здесь: https://stackoverflow.com/questions/793 ... new-string

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