从三维numpy数组生成一个元组。

Question

如果我有31-d numpy数组a，b，c，并且我想构造一个元组数组，其中每个元组是(ai，bi，ci) (所以每个数组的一个列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)]

Python3只需要list调用。

正如@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

收益应该随着数组的大小而增加。