如何在Numba njit函数中创建以UniTuple为键的类型化字典？

• 报错的核心并非UniTuple类型不支持作为Numba类型化字典的键，而是nopython模式对函数内部可访问的对象做了严格的白名单校验：numba.types下的泛型类型构造器（如UniTuple）不在njit函数内部允许访问的属性列表中，因此访问时会触发Unknown attribute 'UniTuple'的类型检查错误。
• 在模块层级实例化字典可正常运行，是因为模块级代码直接由Python解释器执行，完全不受nopython模式的访问规则约束。
• 把键类型换成float64可正常运行，是因为numba.float64是Numba提前注册到nopython全局白名单的基础标量常量，不需要通过访问numba.types模块获取，因此不会触发校验报错。

所有方案都绕开了在njit函数内部直接访问numba.types模块的问题，且均支持以3元素浮点元组作为字典键。

方案1：提前在普通Python环境构造类型常量 #

将类型定义放在njit函数外部的普通代码段，编译期直接将类型作为常量嵌入函数逻辑：

import numba

# 普通Python环境下构造类型，不受nopython模式规则约束
TUPLE_KEY_TYPE = numba.types.UniTuple(numba.float64, 3)
VALUE_TYPE = numba.float64

@numba.njit
def foo():
    local_dict = numba.typed.Dict.empty(
        key_type=TUPLE_KEY_TYPE,
        value_type=VALUE_TYPE,
    )
    # 验证存取逻辑
    local_dict[(1.0, 2.0, 3.0)] = 4.0
    return local_dict[(1.0, 2.0, 3.0)]

if __name__ == '__main__':
    print(foo())  # 输出4.0

方案2：在njit函数内部用typeof自动推导类型 #

numba.typeof是nopython模式白名单内允许直接调用的函数，可通过临时样例值自动推导对应的Numba类型，无需手动构造泛型类型：

import numba

@numba.njit
def foo():
    # 构造和目标键结构一致的临时样例值
    sample_key = (0.0, 0.0, 0.0)
    local_dict = numba.typed.Dict.empty(
        key_type=numba.typeof(sample_key),
        value_type=numba.float64,
    )
    local_dict[(1.0, 2.0, 3.0)] = 4.0
    return local_dict[(1.0, 2.0, 3.0)]

if __name__ == '__main__':
    print(foo())  # 输出4.0

补充说明：3元素浮点元组对应的UniTuple类型本身是完全支持作为Numba类型化字典键的，其哈希规则与Python原生元组完全一致，不存在类型兼容性问题。

内容的提问来源于stack exchange，提问作者Louis Saglio