Pythonnumpy提取矩陣的某一行或某一列的實例

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 提取矩陣的某一行或某一列的實例就是小編分享給大家的全部內容了，希望能給大家一個參考