Объединение вложенных списков в столбцах Polars в строкиPython

Программы на Python
Ответить
Anonymous
 Объединение вложенных списков в столбцах Polars в строки

Сообщение Anonymous »

Как вы, возможно, заметили из других моих вопросов, прямо сейчас я перехожу от панд к полярам. У меня есть полярный df с по-разному вложенными списками, например:
┌────────────────────────────────────┬────────────────────────────────────┬─────────────────┬──────┐
│ col1 ┆ col2 ┆ col3 ┆ col4 │
│ --- ┆ --- ┆ --- ┆ --- │
│ list[list[str]] ┆ list[list[str]] ┆ list[str] ┆ str │
╞════════════════════════════════════╪════════════════════════════════════╪═════════════════╪══════╡
│ [["a", "a"], ["b", "b"], ["c", "c"]┆ [["a", "a"], ["b", "b"], ["c", "c"]┆ ["A", "B", "C"] ┆ 1 │
│ [["a", "a"]] ┆ [["a", "a"]] ┆ ["A"] ┆ 2 │
│ [["b", "b"], ["c", "c"]] ┆ [["b", "b"], ["c", "c"]] ┆ ["B", "C"] ┆ 3 │
└────────────────────────────────────┴────────────────────────────────────┴─────────────────┴──────┘

Теперь я хочу соединить списки наизнанку, используя разные разделители, чтобы добиться этого:
┌─────────────┬─────────────┬───────┬──────┐
│ col1 ┆ col2 ┆ col3 ┆ col4 │
│ --- ┆ --- ┆ --- ┆ --- │
│ str ┆ str ┆ str ┆ str │
╞═════════════╪═════════════╪═══════╪══════╡
│ a+a-b+b-c+c ┆ a+a-b+b-c+c ┆ A-B-C ┆ 1 │
│ a+a ┆ a+a ┆ A ┆ 2 │
│ b+b-c+c ┆ b+b-c+c ┆ B-C ┆ 3 │
└─────────────┴─────────────┴───────┴──────┘

Я делаю это с помощью map_elements и цикла for, но считаю, что это крайне неэффективно. Есть ли собственный способ справиться с этим в Polars?
Вот мой код:
import polars as pl

df = pl.DataFrame({"col1": [[["a", "a"], ["b", "b"], ["c", "c"]], [["a", "a"]], [["b", "b"], ["c", "c"]]],
"col2": [[["a", "a"], ["b", "b"], ["c", "c"]], [["a", "a"]], [["b", "b"], ["c", "c"]]],
"col3": [["A", "B", "C"], ["A"], ["B", "C"]],
"col4": ["1", "2", "3"]})

nested_list_cols = ["col1", "col2"]
list_cols = ["col3"]

for col in nested_list_cols:
df = df.with_columns(pl.lit(df[col].map_elements(lambda listed: ['+'.join(element) for element in listed], return_dtype=pl.List(pl.String))).alias(col)) # is the return_dtype always pl.List(pl.String)?
for col in list_cols + nested_list_cols:
df = df.with_columns(pl.lit(df[col].list.join(separator='-')).alias(col))


Подробнее здесь: https://stackoverflow.com/questions/791 ... to-strings
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

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