从3个1-D Numpy阵列中制作元组

Question

如果我有3个1-d numpy Arrays a，b，c，我想在每个元组中构造一个元组(a [i]，b [i]，c[i]((每个数组A，B，C(的一列(，最快，最有效的方法是什么(无迭代(？

示例：

a = [1,2,3]
b = [4,5,6]
c = [7,8,9]
#resulting tuple array would be
tup = [(1,4,7),(2,5,8),(3,6,9)]

Answer 1

您可以简单地zip他们：

>>> a = np.array([1,2,3])
>>> b = np.array([4,5,6])
>>> c = np.array([7,8,9])
>>> 
>>> list(zip(a,b,c))
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]

list仅需要Python 3。

正如@pirsquared注意，首先将阵列转换为列表：

，您可以使用zip加快一点

In [11]: %%timeit
    ...: list(zip(a,b,c))
    ...:
1000000 loops, best of 3: 1.91 µs per loop
In [12]: %%timeit
    ...: list(zip(*map(np.ndarray.tolist, [a, b, c])))
    ...: 
1000000 loops, best of 3: 1.75 µs per loop

收益应随阵列的大小而增加。