当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python dask.array.nancumsum用法及代码示例


用法:

dask.array.nancumsum(x, axis, dtype=None, out=None, *, method='sequential')

返回给定轴上数组元素的累积总和,将非数字 (NaN) 视为零。当遇到 NaN 并且前导 NaN 被零替换时,累积和不会改变。

此文档字符串是从 numpy.nancumsum 复制的。

可能存在与 Dask 版本的一些不一致之处。

Dask 添加了一个额外的仅关键字参数 method

方法{‘sequential’, ‘blelloch’},可选

选择用于执行 cumsum 的方法。默认为‘sequential’。

  • ‘sequential’ 在当前块之前执行每个先前块的cumsum。
  • ‘blelloch’ 是一个 work-efficient 并行 cumsum。它通过以下方式暴露了并行性

    首先取每个块的总和并通过二叉树组合总和。根据工作负载、调度程序和硬件,此方法可能更快或更节省内存。更多的基准测试是必要的。

对于 all-NaN 或空的切片返回零。

参数

a数组(在 Dask 中不支持)

输入数组。

axis整数,可选

计算累积和的轴。默认值 (None) 是计算展平数组上的 cumsum。

dtypedtype,可选

返回数组的类型和对元素求和的累加器的类型。如果未指定 dtype ,则默认为 a 的 dtype,除非 a 具有精度小于默认平台整数的整数 dtype。在这种情况下,将使用默认平台整数。

outndarray,可选

用于放置结果的替代输出数组。它必须具有与预期输出相同的形状和缓冲区长度,但如果需要,类型将被强制转换。有关更多详细信息,请参阅输出类型确定。

返回

nancumsum数组。

除非指定了out,否则将返回一个保存结果的新数组,并在其中返回它。如果 axis 不是 None 或 a 是一维数组,则结果的大小与 a 相同,形状与 a 相同。

例子

>>> np.nancumsum(1)  
array([1])
>>> np.nancumsum([1])  
array([1])
>>> np.nancumsum([1, np.nan])  
array([1.,  1.])
>>> a = np.array([[1, 2], [3, np.nan]])  
>>> np.nancumsum(a)  
array([1.,  3.,  6.,  6.])
>>> np.nancumsum(a, axis=0)  
array([[1.,  2.],
       [4.,  2.]])
>>> np.nancumsum(a, axis=1)  
array([[1.,  3.],
       [3.,  3.]])

相关用法


注:本文由纯净天空筛选整理自dask.org大神的英文原创作品 dask.array.nancumsum。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。