将不同大小数组的列表传递给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的元素是更复杂的数组，但这只是一个例子(。我怎样才能做到这一点？我知道麻木不喜欢列表。

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

Answer 1

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

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

另一个选项是键入列表和键入dicts:

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