拼合3 d numpy的陣列

Question

如何扁平化這樣的：

b = np.array([ 
    [[1,2,3], [4,5,6], [7,8,9]], 
    [[1,1,1],[2,2,2],[3,3,3]] 
]) 

到：

c = np.array([ 
    [1,2,3,4,5,6,7,8,9], 
    [1,1,1,2,2,2,3,3,3] 
]) 

Niether這些工作：

c = np.apply_along_axis(np.ndarray.flatten, 0, b) 
c = np.apply_along_axis(np.ndarray.flatten, 0, b) 

剛剛返回相同的陣列。

這將是很好的扁平化這個地方。

Answer 1

這將做的工作：

c=b.reshape(len(b),-1) 

然後c是

array([[1, 2, 3, 4, 5, 6, 7, 8, 9], 
     [1, 1, 1, 2, 2, 2, 3, 3, 3]]) 

Answer 2

您可以完全壓平，然後重塑：

c = b.flatten().reshape(b.shape[0],b.shape[1]*b.shape[2]) 

輸出

array([[1, 2, 3, 4, 5, 6, 7, 8, 9], 
     [1, 1, 1, 2, 2, 2, 3, 3, 3]]) 

Answer 3

所以，你總是可以只使用重塑：

b.reshape((2,9)) 
array([[1, 2, 3, 4, 5, 6, 7, 8, 9], 
     [1, 1, 1, 2, 2, 2, 3, 3, 3]])