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
正常得出结果!