我目前有一个函数,可以使用convolve
和hstack
计算出一维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)完成上述操作,以实现相同的功能?
预先感谢您的任何建议或见识。