当输入和返回是字典时,如何使用 numba 加速功能?
我熟悉将 numba 用于接受数字和返回数组的函数,如下所示:
@numba.jit('float64[:](int32,int32)',nopython=True)
def f(a, b):
# returns array 1d array
Run Code Online (Sandbox Code Playgroud)
现在我有一个接受并返回字典的函数。我如何在这里申请 numba?
def collocation(aeolus_data,val_data):
...
return sample_aeolus, sample_valdata
Run Code Online (Sandbox Code Playgroud)