Dask相当于numpy(convolve + hstack)?

时间:2018-11-14 03:31:14

标签: dask

我目前有一个函数,可以使用convolvehstack计算出一维numpy数组(向量)上的滑动和。我想使用dask创建一个等效的函数,但是到目前为止,我尝试过的各种方法还没有解决。

我想做的是计算一个数组的n个数字的“和”,除非任何数字都是NaN,在这种情况下,总和也应该是NaN。结果的第(n-1)个元素也应为NaN,因为没有假定环绕/后移。

例如:

Collection<Object>

input vector: [3, 4, 6, 2, 1, 3, 5, np.NaN, 8, 5, 6] 
n: 3
result: [NaN, NaN, 13, 12, 9, 6, 9, NaN, NaN, NaN, 19]

我目前使用numpy函数实现的功能:

input vector: [1, 5, 7, 2, 3, 4, 9, 6, 3, 8]
n: 4
result: [NaN, NaN, NaN, 15, 17, 16, 18, 22, 22, 26]

如何使用dask数组API(和/或dask_image.ndfilters)完成上述操作,以实现相同的功能?

预先感谢您的任何建议或见识。

0 个答案:

没有答案