python - 将 Dask 标量转换为整数值(或将其保存到文本文件)

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