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 documentation

中找到一个示例

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 data type object，而不是Python中的实际新类型

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

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

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

2 个答案: