如何在 numba 中声明其值为嵌套列表的类型化字典

Question

您好，我知道 numba 有一个 numba.typed.Dict ，它是一个类型化的字典，但我的字典的值是一个列表列表，字典中的每个键可能具有不同长度的外部列表； 每个内部列表也可能有不同的长度； 内部列表的类型是一对 float32 。 我想知道 numba 是否支持这种结构以及如何声明它？

字典示例如下

{
    “Seattle”: [[[1.2, 3.5], [4.5, 6.7]]],
    “New York”:[ [[5.7, 22.4], [3.5, 7.8], [6.9, 4.1]], [[2.3, 4.5]]],
    “Boston”: [[[2.4, 5.0]]]
}


region_polygons = Dict.empty(
key_type=types.unicode_type,
value_type=  <------ try to figure out this )

Answer 1

我们可以利用支持嵌套的类型列表。 我正在使用List(lsttype=...)手动指定官方文档不幸没有显示的类型，但请记住，类型列表目前是一个实验性功能（v 0.47），可能会在以后的版本中发生变化.

>>> from numba.typed import Dict, List
>>> from numba.types import float32, unicode_type, ListType

>>> py_dct = {
...     "Seattle": [[[1.2, 3.5], [4.5, 6.7]]],
...     "New York":[ [[5.7, 22.4], [3.5, 7.8], [6.9, 4.1]], [[2.3, 4.5]]],
...     "Boston": [[[2.4, 5.0]]]
... }

>>> nested_3 = ListType(ListType(ListType(float32)))
>>> nested_2 = ListType(ListType(float32))
>>> nested_1 = ListType(float32)

>>> nb_dct = Dict.empty(
...     key_type=unicode_type,
...     value_type=nested_3,
... )

>>> for city, lst in py_dct.items():
...     out_lst = List(lsttype=nested_3)
...     for outer in lst:
...         mid_lst = List(lsttype=nested_2)
...         for middle in outer:
...             in_lst = List(lsttype=nested_1)
...             for inner in middle:
...                 in_lst.append(inner)
...             mid_lst.append(in_lst)
...         out_lst.append(mid_lst)
...     nb_dct[city] = out_lst

>>> nb_dct['Seattle']
ListType[ListType[ListType[float32]]]([[[1.2000000476837158, 3.5], [4.5, 6.699999809265137]]])

>>> nb_dct['New York']
ListType[ListType[ListType[float32]]]([[[5.699999809265137, 22.399999618530273], [3.5, 7.800000190734863], [6.900000095367432, 4.099999904632568]], [[2.299999952316284, 4.5]]])

>>> nb_dct['Boston']
ListType[ListType[ListType[float32]]]([[[2.4000000953674316, 5.0]]])