代码之家  ›  专栏  ›  技术社区  ›  user9885031

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

  •  0
  • user9885031  · 技术社区  · 8 年前

    我用 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()

    谢谢

    1 回复  |  直到 8 年前
        1
  •  9
  •   mdurant    8 年前

    .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