3d Numpy array to 2d

前端 未结 1 838
日久生厌
日久生厌 2020-11-30 02:25

I have a 3d matrix like this

arange(16).reshape((4,2,2))
array([[[ 0,  1],
        [ 2,  3]],

        [[ 4,  5],
        [ 6,  7]],

        [[ 8,  9],
             


        
相关标签:
1条回答
  • 2020-11-30 03:00
    In [27]: x = np.arange(16).reshape((4,2,2))
    
    In [28]: x.reshape(2,2,2,2).swapaxes(1,2).reshape(4,-1)
    Out[28]: 
    array([[ 0,  1,  4,  5],
           [ 2,  3,  6,  7],
           [ 8,  9, 12, 13],
           [10, 11, 14, 15]])
    

    I've posted more general functions for reshaping/unshaping arrays into blocks, here.

    0 讨论(0)
提交回复
热议问题