Вменение и добавление строк в DataFrame с использованием экспрессии Polars ⇐ Python

[Anonymous]

Я имею данные о данных с неполными значениями, как показано ниже - в частности, с соответствующими годами, и я хотел бы сделать его квадратным (то есть все три cust_id, чтобы правильно вменять значения для возраста за все три года, то есть я хочу повернуть это: < /p>

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

df = pl.DataFrame({
"cust_id": [1, 2 ,2, 2, 3, 3],
"year": [2000,1999,2000,2001,1999,2001],
"cust_age": [21,31,32,33,44,46]

})
< /code>
в это: < /p>
df = pl.DataFrame({
"cust_id": [1, 1, 1, 2 ,2, 2, 3, 3, 3 ],
"year": [2000,1999,2000,2001,1999,2001],
"cust_age": [20,21,22,31,32,33,44,45,46]

})

Я знаю, что могу итеративно создать новые данные DataFrames, а затем присоединиться или сдвинуть значения, и потенциально использовать Apply и/или eval_map, но я надеялся найти что -то быстрое и идиоматическое для поляров. Заранее спасибо за помощь!

Подробнее здесь: https://stackoverflow.com/questions/796 ... xpressions