Python Polars – создать последовательность в списке из целого числаPython

Программы на Python
Ответить
Anonymous
 Python Polars – создать последовательность в списке из целого числа

Сообщение Anonymous »

Учитывая, что:

Использование map_elements намного медленнее, чем собственный API выражений.

Как я могу использовать собственный API выражений Polars для создания нового столбца со списком, содержащим все целые числа от 1 до номера другого столбца?
Это означает, что исходя из этого:
shape: (3, 1)
┌─────┐
│ Qty │
│ --- │
│ i64 │
╞═════╡
│ 1 │
│ 2 │
│ 3 │
└─────┘

На это:
shape: (3, 2)
┌─────┬───────────┐
│ Qty ┆ list │
│ --- ┆ --- │
│ i64 ┆ list[i64] │
╞═════╪═══════════╡
│ 1 ┆ [1] │
│ 2 ┆ [1, 2] │
│ 3 ┆ [1, 2, 3] │
└─────┴───────────┘

Моя текущая попытка использования map_elements:
# Import package.
import polars as pl

# Create the DataFrame.
df = pl.DataFrame({'Qty': [1, 2, 3]})

# Add the 'List' column.
df = df.with_columns(
pl.col('Qty').map_elements(lambda x: [i+1 for i in range(x)]).alias('List')
)


Подробнее здесь: https://stackoverflow.com/questions/768 ... om-integer
Ответить

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

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

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

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

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