Взвешенная сумма столбца в кадре данных Polars ⇐ Python

[Anonymous]

У меня есть кадр данных Polars, и я хочу вычислить взвешенную сумму определенного столбца, а веса — это просто последовательность положительных целых чисел, например, 1, 2, 3,...
Например, предположим, что у меня есть следующий фрейм данных.

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

import polars as pl

df = pl.DataFrame({"a": [2, 4, 2, 1, 2, 1, 3, 6, 7, 5]})

Мне нужен такой результат

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

218 (= 2*1 + 4*2 + 2*3 + 1*4 + ... + 7*9 + 5*10)

Как я могу добиться этого, используя только общие полярные выражения? (Причина, по которой я хочу использовать только полярные выражения для решения проблемы, заключается в соображениях скорости)< /p>
Примечание. Это простой пример, в котором всего 10 чисел, но в целом высота кадра данных может быть любым положительным числом.
Спасибо за помощь.

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