0.44的numba会用numba.typed.Dict()替换掉原来python自带的dict类型。

举个栗子:

#定义一个numba支持的dict类型
numba_dict= Dict.empty(
    key_type = types.UniTuple(types.int64,2),#numba_dict支持元祖类型
    value_type = types.int64[:],
)

#定义一个普通的dict类型
dict = {(3,55):np.array([1480, 1527, 1528, 1565, 1566]),(3,66):np.array([1480, 1527, 1528, 1565, 1566])}

#普通类型dict赋值给numba支持的dict
for i in dict:
    list = []
    for j in dict[i]:
        list.append(j)
    numba_dict[i] = np.array(list).astype(np.int64)


@njit
def sum_dict(dict_test):
    sum = 0
    for i in dict_test:
        count= dict_test[i].sum()
        sum+=count

    return sum

 下面来验证一下numba对于普通dict 和 numba_dict 的支持情况:

print(sum_test(dict))



#发现报错,报错信息如下
def sum_dict(dict_test):
    sum = 0
    ^
This error may have been caused by the following argument(s):
- argument 0: cannot determine Numba type of <class 'dict'>
print(sum_dict(numba_dict))

15332



正常得出结果!

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐