В Polars – как мне проще добавить две строки в конце кадра данных: одну для общего количества каждого столбца и одну для ⇐ Python

[Anonymous]

Например

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

import polars as pl
df=pl.DataFrame({
"Col Ind": ['A','B','C','D','E'],
"Col A": [1,2,3,4,5],
"Col B": [2,4,6,8,10],
"Col C": [1,3,5,7,9],
"Col D": [5,4,3,2,1] })

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

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

first_col = df.select("Col Ind").to_series().append(pl.Series("temp", ["Total", "Percentage"]))
df = df.drop("Col Ind")
cols = df.columns
expr = df.select(pl.sum(cols))
rowlist = list(expr.row(0))
full = sum(rowlist)
pc_row = []
for n in range(len(rowlist)):
pc_row.append(int(rowlist[n] /full *100))

pc_dict = dict(zip(cols, pc_row))
pc_df = pl.DataFrame(pc_dict)
df = pl.concat([df,expr])
df = pl.concat([df,pc_df])
df.insert_column(0, first_col)

Это работает, но кажется, что шагов слишком много.
Есть ли более простой способ?
Спасибо.

Подробнее здесь: https://stackoverflow.com/questions/792 ... me-one-for