import numpy as npa=np.arange(9).reshape(3,3)12
a
Out[31]:
array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])12345
矩阵的某一行
a[1]Out[32]: array([3, 4, 5])12
矩阵的某一列
a[:,1]Out[33]: array([1, 4, 7])12
b=np.eye(3,3)
b
Out[36]:
array([[ 1., 0., 0.],
[ 0., 1., 0.],
[ 0., 0., 1.]])1234567
把矩阵a的第2列赋值给矩阵b的第1列
b[:,0]=a[:,1]
b
Out[38]:
array([[ 1., 0., 0.],
[ 4., 1., 0.],
[ 7., 0., 1.]])