微客导航 » 文章资讯 » Python numpy 提取矩阵的某一行或某一列的实例

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

2023-09-14 14:39:05 364

如下所示：

importnumpyasnp
a=np.arange(9).reshape(3,3)

a
Out[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)

b
Out[36]:
array([[1.,0.,0.],
[0.,1.,0.],
[0.,0.,1.]])

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

b[:,0]=a[:,1]

b
Out[38]:
array([[1.,0.,0.],
[4.,1.,0.],
[7.,0.,1.]])

以上这篇Pythonnumpy提取矩阵的某一行或某一列的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。