python: numpy: 选取矩阵元素的方法

>>> a = np.random.random((3,4))
>>> a
array([[0.890582  , 0.43671675, 0.19531267, 0.13555252],
       [0.17254557, 0.3088476 , 0.36916878, 0.81551741],
       [0.31869139, 0.16168486, 0.58004271, 0.4279148 ]])
>>> a.nonzero()
(array([0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]), array([0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3]))

>>> a[a.nonzero()]   ##方法一,使用矩阵去选择
array([0.890582  , 0.43671675, 0.19531267, 0.13555252, 0.17254557,
       0.3088476 , 0.36916878, 0.81551741, 0.31869139, 0.16168486,
       0.58004271, 0.4279148 ])
>>> a
array([[0.890582  , 0.43671675, 0.19531267, 0.13555252],
       [0.17254557, 0.3088476 , 0.36916878, 0.81551741],
       [0.31869139, 0.16168486, 0.58004271, 0.4279148 ]])
>>> a[1,1]         ##方法二
0.3088475978138986
>>> a[1][1]         ##方法三
0.3088475978138986
>>> a[np.array([[False,False,False,False],[True,True,True,True,],
		[False,False,False,False]])]

array([0.72159605, 0.90488492, 0.60745105, 0.20538722])  ##方法四

 

  • 5
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
numpy中,可以使用下标或者切片来访问矩阵元素。例如,对于一个二维矩阵a,可以使用a[i][j]或a[i,j]来访问第i行第j列的元素。这里的i和j都是从0开始的整数索引。 另外,numpy还支持使用布尔值数组、整数数组和切片对象来进行高级索引。例如,可以使用一个布尔值数组b来选取矩阵a中所有满足条件的元素: ```python import numpy as np a = np.array([[1, 2], [3, 4], [5, 6]]) b = np.array([[True, False], [False, True], [True, False]]) print(a[b]) ``` 输出结果为: ``` [1 4 5] ``` 其中,b中的每个元素表示对应位置的元素是否被选中。在这个例子中,b表示选取第1行第1列和第3行第1列的元素。因此,输出结果为1、4和5。 除了布尔值数组之外,还可以使用整数数组和切片对象来进行高级索引。例如,可以使用一个整数数组c来选取矩阵a中指定位置的元素: ```python c = np.array([0, 1, 0]) print(a[c]) ``` 输出结果为: ``` [[1 2] [3 4] [1 2]] ``` 其中,c中的每个元素表示对应位置的行索引。在这个例子中,c表示选取第1行、第2行和第1行的元素。因此,输出结果为一个新的矩阵,其中第1行和第3行的元素重复出现了。 最后,对于多维矩阵,可以使用多个切片对象来进行索引。例如,可以使用下面的语句来选取矩阵a的第1和第3行,以及第2和第3列的元素: ```python print(a[[0,2]][:,1:]) ``` 输出结果为: ``` [[2] [6]] ``` 其中,a[[0,2]]表示选取第1和第3行的元素,而[:,1:]表示选取第2和第3列的元素。因此,输出结果为2和6。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值