两个例子读懂np.concatenate

>>> a = np.array([[1, 2], [3, 4]])
>>> b = np.array([[5, 6]])               b是一个二维array
>>> np.concatenate((a, b), axis=0)
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> np.concatenate((a, b.T), axis=1)
array([[1, 2, 5],
       [3, 4, 6]])
>>> a = np.array([[1, 2], [3, 4]])                a、b的shape为(2,2),连接第一维就变成(4,2),连接第二维就变成(2,4)
>>> b = np.array([[5, 6], [7, 8]])
>>> np.concatenate((a,b),axis=0)
array([[1, 2],
       [3, 4],
       [5, 6],
       [7, 8]])
>>> np.concatenate((a,b),axis=1)
array([[1, 2, 5, 6],
       [3, 4, 7, 8]])
发布了218 篇原创文章 · 获赞 110 · 访问量 22万+

猜你喜欢

转载自blog.csdn.net/zhuiyunzhugang/article/details/104793227