python - 是否可以在 guvectorize 函数中返回输出数组形状未知的数组

Question

我想知道如果我不知道 numba guvectorize 函数先验的数组大小，如何返回数组。我基本上有一个接收向量的函数，并返回一个数组，其大小将在函数中发生变化。我的问题是我无法先验设置返回数组的形状。

例如:

@guvectorize ([(float64[:],int64[:,:])], '(n),()->(n,2)')
def mysegmentation(Thresholding,StimSegments_time):
    count = 0
    for i in range(Thresholding.shape[0]-1):
        if Thresholding[i]==0. and Thresholding[i+1]==1.:
            StimSegments_time[count,0] = i 
        elif Thresholding[i]==1. and Thresholding[i+1]==0.:
            StimSegments_time[count,1] = i
            count += 1

阈值可以有(1000)个元素，StimSegments_time 可以有(5, 2)或(7, 2)或其他元素(x,2)

Answer 1

我也遇到同样的问题。但我所能提供的帮助是，您不能在装饰器中添加固定数字(即 ->(n,2) 不正确)。

对于这个问题，人们建议在输入中添加一个虚拟数组参数，并在输出中使用其长度，如下所述:

guvectorize not accepting new size variable