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

這些工作無足輕重：

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