In einem Polars DataFrame möchte ich INTs durch eine Liste von INTs ersetzen ⇐ Python

[Anonymous]

Ich habe eine Karte, die ein Diktat ist, das eine int und eine Karten in eine Liste von INTs nimmt. Sollte es einige INTs in der Karte geben, sollte sie stattdessen durch eine Liste von Nullen ersetzt werden.

Code: Select all

import polars as pl

data = {
"user_id": [1, 2, 3],
"book_ids": [[101, 102, 103], [104, 105], [106]]
}

# Create DataFrame
read_history_data = pl.DataFrame(data)

# Mapping dictionary
map = {
101: [1, 2],
102: [3, 4],
103: [5, 6],
104: [7, 8],
105: [9, 10],

106: [11, 12]
}

# Padding value and token length
padding_value = 0
token_length = 4

# Column name
column = "book_ids"

# Function to transform the DataFrame
def transform_read_history_data(read_history_data, map, padding_value, token_length, column):
padded_list = [padding_value for i in range(token_length)]
read_history_data = read_history_data.with_columns(
pl.col(column)
.list.eval(pl.element().replace(map, default=None))
.list.eval(pl.element().fill_null(padded_list))
)
return read_history_data

# Run the function
transformed_data = transform_read_history_data(read_history_data, map, padding_value, token_length, column)

# Print the transformed DataFrame
print(transformed_data)
< /code>
Ich erhalte: < /p>
Traceback (most recent call last):
File "", line 37, in 
File "", line 29, in transform_read_history_data
File "c:\...\.venv\Lib\site-packages\polars\dataframe\frame.py", line 9830, in with_columns
return self.lazy().with_columns(*exprs, **named_exprs).collect(_eager=True)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "c:...\.venv\Lib\site-packages\polars\_utils\deprecation.py", line 93, in wrapper
return function(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^
File "c:\...\.venv\Lib\site-packages\polars\lazyframe\frame.py", line 2224, in collect
return wrap_df(ldf.collect(engine, callback))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
polars.exceptions.ShapeError: argument 2 called 'new' for replace_strict have different lengths (6 != 3)