python - 如何用Numba创建dict类型的类属性？

Question

我想在 Numba 类中使用字典，但我不确定要为类型规范添加什么:

@numba.jitclass([("cnts", ???)])
class Counter:
    def __init__(self):
        self.cnts = numba.typed.Dict.empty(
            key_type=numba.float64, value_type=numba.int64
        )

有什么想法吗？

Answer 1

这对我在 Numba 0.45.1 上有效:

import numba

@numba.jitclass([("cnts", numba.types.DictType(numba.float64, numba.int64))])
class Counter:
    def __init__(self):
        self.cnts = numba.typed.Dict.empty(
            key_type=numba.float64, value_type=numba.int64
        )

    def add(self, x):
        if x not in self.cnts:
            self.cnts[x] = 0
        self.cnts[x] += 1

c = Counter()
c.add(1.1)
c.add(1.1)
c.add(0.5)
print(c.cnts)  # {1.1: 2, 0.5: 1}