如何从Polars DataFrame中获取sum()计算结果值？求更简便方法

你完全不需要转成列表再取值，Polars提供了更直接的方法提取求和后的标量值：

单列求和直接取数值 #

直接对目标列调用sum()后，使用.item()就能拿到Python原生类型的求和结果：

total = df["column_name"].sum().item()
print(total)

多列/分组求和的取值场景 #

如果是通过select()生成的单行求和结果DataFrame，也可以直接用.item()指定列提取，或者转字典取值：

# 提取指定列的求和结果
total = df.select(pl.col("column_name").sum()).item("column_name")

# 多列求和后转字典取值
sum_dict = df.select(pl.col(["col1", "col2"]).sum()).to_dict(as_series=False)
total_col1 = sum_dict["col1"][0]
total_col2 = sum_dict["col2"][0]

这种方式跳过了列表转换的步骤，不仅代码更简洁，在处理大数据集时效率也会更高。

内容的提问来源于stack exchange，提问作者okayama-taro