将 Dask Scalar 转换为整数值（或将其保存到文本文件）

Question

我已经使用dask 计算了

from dask import dataframe
all_data = dataframe.read_csv(path) 
total_sum = all_data.account_balance.sum()

csv 文件有一个名为account_balance 的列。

total_sum 是一个dd.Scalar 对象，似乎很难将其更改为整数。如何获得它的整数版本？或将其保存在包含数字的.txt 文件中也可以。

我也试过total_sum.compute()。

谢谢。

Answer 1

.compute() 确实给你带来了一个实数，正如你在这个例子中看到的那样：

In [18]: import dask.dataframe as dd

In [19]: d = dd.from_pandas(pd.DataFrame({'a': [3,3,3,3]}), npartitions=2)

In [20]: d.a.sum().compute()
Out[20]: 12