今天小编为大家分享Python中numpy按行或列提取矩阵的方法，有需要的小伙伴可以参考一下。

代码：

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

</pre>
a Out[31]: array([[0,1,2], [3,4,5], [6, 7, 8]])
<pre class="brush:js;toolbar:false">

矩阵的某一行

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

</pre>
矩阵的某一列

 

a[:,1] Out[33]:array([1, 4, 7])
<pre class="brush:js;toolbar:false">

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

</pre>
把矩阵a的第2列赋值给矩阵b的第1列

 

b[:,0]=a[:,1] b Out[38]: array([[ 1., 0., 0.], [ 4., 1., 0.], [ 7., 0., 1.]])
<pre class="brush:js;toolbar:false">

以上就是Python中numpy按行或列提取矩阵的方法，学会的小伙伴可以自己尝试下。