将不同大小的数组列表传递给Numba函数

Question

我有一个预先计算好的大小不同的数组列表，我想将其传递给Numba函数。

from numba import jit
import numpy as np

@jit(nopython=True)
def go_fast(a, b): 
    ...
    return output

a = np.arange(100).reshape(10, 10)
b=[np.arange(4),np.arange(9)]

（实际上，b的元素是更复杂的数组，但这只是一个示例）。我该怎么做？我知道numba不喜欢列表。

一种方法是将b变成具有填充的高维数组，但是提取实元素将需要循环，这是不理想的。有更好的方法吗？

Answer 1

新版本似乎支持列表列表：

https://numba.pydata.org/numba-doc/dev/reference/pysupported.html

另一个选项是类型列表和类型字典：

https://numba.pydata.org/numba-doc/dev/reference/pysupported.html#dict https://numba.pydata.org/numba-doc/dev/reference/pysupported.html#list