Numpy中的矩阵合并

时间:2023-03-08 17:58:30
Numpy中的矩阵合并

列合并/扩展:np.column_stack()

行合并/扩展:np.row_stack()

>>> import numpy as np
>>> a = np.arange(9).reshape(3,-1)
>>> a array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
>>> b = np.arange(10, 19).reshape(3, -1)
>>> b array([[10, 11, 12],
[13, 14, 15],
[16, 17, 18]])
>>> top = np.column_stack((a, np.zeros((3,3))))
>>> top array([[ 0., 1., 2., 0., 0., 0.],
[ 3., 4., 5., 0., 0., 0.],
[ 6., 7., 8., 0., 0., 0.]])
>>> bottom = np.column_stack((np.zeros((3,3)), b))
>>> bottom array([[ 0., 0., 0., 10., 11., 12.],
[ 0., 0., 0., 13., 14., 15.],
[ 0., 0., 0., 16., 17., 18.]])
>>> np.row_stack((top, bottom)) array([[ 0., 1., 2., 0., 0., 0.],
[ 3., 4., 5., 0., 0., 0.],
[ 6., 7., 8., 0., 0., 0.],
[ 0., 0., 0., 10., 11., 12.],
[ 0., 0., 0., 13., 14., 15.],
[ 0., 0., 0., 16., 17., 18.]])