Вот наивное решение того, что я хочу сделать, используя map_elements
. Как я могу сделать это, используя только функции Polars?
import polars as pl
# Create a DataFrame with a column containing lists of strings
df = pl.DataFrame({
"list_of_strings": [["a", "b", "c"], ["d", "e", "f"], ["g", "h", "i"]]
})
# Define a function to concatenate lists of strings into a single string
def concatenate_list_of_strings(lst):
return "".join(lst)
# Apply the function to the DataFrame
df = df.with_column(
pl.col("list_of_strings").map_elements(concatenate_list_of_strings, return_dtype=pl.String).alias("concatenated_string")
)
print(df)
Как уже упоминалось в комментариях, в собственном API выражений Polars есть pl.Expr.list.join для объединения всех строковых элементов в подсписке с разделителем между ними.
df.with_columns(
pl.col("list_of_strings").list.join("")
)
shape: (3, 1)
┌─────────────────┐
│ list_of_strings │
│ --- │
│ str │
╞═════════════════╡
│ abc │
│ def │
│ ghi │
└─────────────────┘
pl.col(..).list.join("")
?