python - 如何从 Numba prange 存储结构化数据

Question

我有这样的功能：

@jit(nopython=True, nogil=True, parallel=True)
def parallelLoop():

    X = [None] * 10

    for i in prange(10):
        X[i] = foo()

wherefoo()返回一个可变大小的 numpy 结构化数组。运行它会给我错误：

numba.errors.TypingError: Failed at nopython (nopython frontend)
invalid setitem with value of (list(float64), float64, float64, float64, int64, array(float64, 2d, C)) to element of none

如何在 Numba 中存储并行线程的结果，其中结果是不同大小的结构化数组？

标签： pythonnumpynumba