從3個1-d numpy數組中創建一個元組

Question

如果我有3個1-d numpy數組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)]

只有Python 3才需要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

增益應隨着陣列的大小而增加。