如果我有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)]
如果我有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)]
你可以簡單地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
的收益應與數組的大小增加。
**如果**'a','b'和'c'是'np.array's ...你可以通過'list(zip(* map(np.ndarray.tolist, [a,b,c])))' – piRSquared
@piRSquared哦不錯,只是做了時間。 –
那些不是numpy數組,它們是* lists *。 –