小编Ann*_*erl的帖子

使用带有 numba njit 函数的字典

当输入和返回是字典时,如何使用 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)

python dictionary numba

15
推荐指数
1
解决办法
1万
查看次数

标签 统计

dictionary ×1

numba ×1

python ×1