如何连接 numba 列表

Question

我用 Numba int32 列表替换了 Python 列表，但它的串联行为与预期不符。

例如 Python：如果 a = [1,2,3] 且 b = [4,5,6] 那么 a += b 给出 [1, 2, 3, 4, 5 , 6].

另一方面，在 Numba 中：如果 a = numba.int32([1,2,3]) 和 b = numba.int32([4,5,6]) 那么 a += b 给出 array([5, 7, 9], dtype=int32) .

有没有一种方法可以像我们在 Python 中那样轻松地连接 Numba 列表？还是编写循环遍历两个数组并创建另一个数组的函数的唯一解决方案？

谢谢。

Answer 1

import numpy as np
a = numba.int32([1,2,3])
b = numba.int32([4,5,6])
c = np.concatenate((a,b))

Answer 2

使用list()

a = numba.int32(list(a)+list(b))

How to concatenate numba lists