2017-06-01 66 views
1

如果我有3個1-d numpy數組a,b,c和我想要構造一個元組數組,其中每個元組都是(a [i],b [i] ,c [i])(所以每個數組a,b,c的列)),這樣做的最快和最有效的方式是什麼(沒有迭代)?從3個1-n numpy數組構成一個元組

例如:

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)]

+1

那些不是numpy數組,它們是* lists *。 –

回答

3

你可以簡單地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ñ OTES,你可以去稍快一點與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 

的收益應與數組的大小增加。

+1

**如果**'a','b'和'c'是'np.array's ...你可以通過'list(zip(* map(np.ndarray.tolist, [a,b,c])))' – piRSquared

+0

@piRSquared哦不錯,只是做了時間。 –

相關問題