How to merge 2 numpy arrays?

后端 未结 3 556
轮回少年
轮回少年 2021-01-25 09:02

I feel like there is some documentation I am missing, but I can\'t find anything on this specific example - everything is just about concatenating or stacking arrays.

I

相关标签:
3条回答
  • 2021-01-25 09:31

    If you have two-dimensional matrices, you can use numpy.dstack():

    z = np.dstack((x, y))
    
    0 讨论(0)
  • 2021-01-25 09:36
    In [39]: z = np.concatenate((x[...,None], y[...,None]), axis=2)
    
    In [40]: z
    Out[40]: 
    array([[[ 1,  7],
            [ 2,  8],
            [ 3,  9]],
    
           [[ 4, 10],
            [ 5, 11],
            [ 6, 12]]])
    
    0 讨论(0)
  • 2021-01-25 09:54

    Sounds like the function you're looking for is stack(), using it to stack along the 3rd dimension.

    import numpy as np
    
    x = np.asarray([[1,2,3],[4,5,6]])
    y = np.asarray([[7,8,9],[10,11,12]])
    z = np.stack((x, y), 2)
    
    0 讨论(0)
提交回复
热议问题