python - 如何从 numba 结构类型创建结构？

Question

我可以定义一个 numba 结构类型：

from numba import struct, int32

my_struct_type = struct([('value_a', int32), ('value_b', int32)])

现在我有了类型，如何创建一个实际的结构？

Answer 1

numba.struct不再存在，您现在可以简单地numpy使用自定义复杂的标准数组传递dtype....

import numpy as np # version 1.10.1
import numba # version 0.22.1

@numba.jit(nopython=True)
def sum_basic_array(a):
    ret = 0
    for aa in a:
        ret += aa
    return ret

@numba.jit(nopython=True)
def sum_struct_array(a):
    ret = 0
    for aa in a:
        ret += aa.sub_0
    return ret

x_basic = np.arange(20000*3, dtype=np.uint32)
x_struct = x_basic.view(dtype=np.dtype([('sub_0', np.int32),
                                        ('sub_1', np.float64)]))

%timeit sum_basic_array(x_basic) # 18 µs
%timeit sum_struct_array(x_struct) # 8 µs
%timeit sum_struct_array(x_struct.view(np.recarray)) # 40 µs

请注意，它的sum_struct_array迭代次数是原来的三分之一，因此它更快也就不足为奇了。

Answer 2

可以在Numba 文档中找到一个示例

record_type = struct([('x', double), ('y', double)])
record_dtype = record_type.get_dtype()
a = numpy.array([(1.0, 2.0), (3.0, 4.0)], dtype=record_dtype)

您创建的结构是一个 numpy数据类型对象，而不是 Python 中的实际新类型

对于您的代码示例，这将是：

my_struct_type = struct([('value_a', int32), ('value_b', int32)])
my_struct = np.array([(1, 2)], dtype=my_struct_type)