连接3D Numpy Arrary

时间:2018-04-02 04:55:29

标签: python-3.x numpy

有没有办法在不使用for循环的情况下实现与以下代码相同的输出?

A:

  [[[ 1  2  3]
    [ 4  5  6]]

   [[ 7  8  9]
    [10 11 12]]]

输出:

 [[ 0  1  2  3  4  5  6  7  8  9 10 11]
  [12 13 14 15 16 17 18 19 20 21 22 23]]

代码:

A=np.arange(24).reshape(2,3,4)

v= []
for i in range(A.shape[0]):
    v.append(np.concatenate(A[i]))

1 个答案:

答案 0 :(得分:2)

再做一次重塑!

v = A.reshape(A.shape[0], -1)

将来使用Python REPL进行实验可能会有所帮助。