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

来源:互联网 发布:linux查看电池电量 编辑:程序博客网 时间:2024/05/17 08:09
import numpy as npa=np.arange(9).reshape(3,3)
aOut[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)bOut[36]: array([[ 1.,  0.,  0.],       [ 0.,  1.,  0.],       [ 0.,  0.,  1.]])

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

b[:,0]=a[:,1]bOut[38]: array([[ 1.,  0.,  0.],       [ 4.,  1.,  0.],       [ 7.,  0.,  1.]])
阅读全文
6 0
原创粉丝点击