combining 2D arrays to 3D arrays

前端 未结 6 769
北恋
北恋 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条回答
  •  时光取名叫无心
    2021-01-03 11:11

    I think I got something that works :

    >>> print np.hstack([A[:, None, :], B[:, None, :], C[:, None, :]])
    [[[ 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]]]
    

提交回复
热议问题