我可以定义一个 numba 结构类型:
from numba import struct, int32
my_struct_type = struct([('value_a', int32), ('value_b', int32)])
既然我有了类型,我该如何创建一个实际的结构?
最佳答案
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
执行的迭代次数是原来的三分之一,因此它更快也就不足为奇了。
关于python - 如何从 numba 结构类型创建结构?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/19605757/