combining 2D arrays to 3D arrays

前端 未结 6 742
北恋
北恋 2021-01-03 10:32

Hello I have 3 numpy arrays as given below.

>>> print A
[[ 1.  0.  0.]
 [ 3.  0.  0.]
 [ 5.  2.  0.]
 [ 2.  0.  0.]
 [ 1.  2.  1.]]
>>> pri         


        
6条回答
  •  -上瘾入骨i
    2021-01-03 11:16

    A solution using numpy dstack:

    >>> import numpy as np
    >>> np.dstack((a,b,c)).swapaxes(1,2)
    array([[[ 1,  0,  0],
            [ 5,  9,  9],
            [ 0,  0,  0]],
    
           [[ 3,  0,  0],
            [37,  8,  9],
            [ 0,  6,  0]],
    
           [[ 5,  2,  0],
            [49,  8,  3],
            [ 1,  4,  6]],
    
           [[ 2,  0,  0],
            [ 3,  3,  1],
            [ 6,  2,  0]],
    
           [[ 1,  2,  1],
            [ 4,  4,  5],
            [ 0,  5,  4]]])
    

提交回复
热议问题