欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-09-06 21:56:24
如下所示: import numpy as np a=np.arange(9).reshape(3,3) a Out[31]: array(...

如下所示:

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 提取矩阵的某一行或某一列的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。