Python中numpy按行或列提取矩阵的示例

这篇文章将为大家详细讲解有关Python中numpy按行或列提取矩阵的示例，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

代码：

import numpy as np
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.]])

关于Python中numpy按行或列提取矩阵的示例就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。