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 执行的迭代次数是原来的三分之一，因此它更快也就不足为奇了。