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