Кадр данных Polars: столбцы списков [некоторые типы данных] в столбцы (некоторые типы данных) ⇐ Python

[Anonymous]

есть фрейм данных Polars. Я проделывал некоторые манипуляции, после которых значения столбцов представляли собой списки некоторых типов данных, таких как список str или список int. Я хочу, чтобы эти столбцы были str или int. в этих списках есть только одно значение, так что это должно быть легко, верно? Как мне это сделать
Это лучший способ сделать это?

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

import polars as pl

# Sample DataFrame with multiple columns
df = pl.DataFrame({
"col1": [[1], [2], [3]],  # list of int
"col2": [["a"], ["b"], ["c"]],  # list of str
"col3": [[4], [5], [6]]  # another list of int
})

# List of columns to convert
columns_to_convert = ["col1", "col2", "col3"]

# Apply the transformation to multiple columns
df = df.with_columns(
pl.col(col).list.first().alias(col) for col in columns_to_convert
)

print(df)

Выход:

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

shape: (3, 3)
┌──────┬──────┬──────┐
│ col1 ┆ col2 ┆ col3 │
│ ---  ┆ ---  ┆ ---  │
│ i64  ┆ str  ┆ i64  │
╞══════╪══════╪══════╡
│ 1    ┆ a    ┆ 4    │
│ 2    ┆ b    ┆ 5    │
│ 3    ┆ c    ┆ 6    │
└──────┴──────┴──────┘

Совместимо ли это с lazyframe?

Подробнее здесь: https://stackoverflow.com/questions/785 ... -datatypes