Python numpy 提取矩阵的某一行或某一列

import numpy as npa=np.arange(9).reshape(3,3)

aOut[31]: array([[0, 1, 2],[3, 4, 5],[6, 7, 8]])

矩阵的某一行

a[1]Out[32]: array([3, 4, 5])

矩阵的某一列

a[:,1]Out[33]: array([1, 4, 7])

b=np.eye(3,3)bOut[36]: array([[ 1., 0., 0.],[ 0., 1., 0.],[ 0., 0., 1.]])

把矩阵a的第2列赋值给矩阵b的第1列

b[:,0]=a[:,1]bOut[38]: array([[ 1., 0., 0.],[ 4., 1., 0.],[ 7., 0., 1.]])