Numpy怎样实现返回指定位置索引

Admin 2022-07-19 群英技术资讯 740 次浏览

这篇文章主要介绍“Numpy怎样实现返回指定位置索引”,有一些人在Numpy怎样实现返回指定位置索引的问题上存在疑惑,接下来小编就给大家来介绍一下相关的内容,希望对大家解答有帮助,有这个方面学习需要的朋友就继续往下看吧。


先上代码,主要语句为np.where(b[c]==1),

详细解释如下:

import numpy as np 
b = np.array([[-2,-3,0,0,0,6,4,1],[88,1,0,0,0,6,4,2],[99,6,0,0,1,6,4,2]]) # 三行八列的数组b
print('b\n',b)
 
c = np.array([2,0])     # c表示指定行
print('b[c]\n',b[c])    # b[c]返回 数组b的指定行  这里依次返回了b的下标为2和0的行
 
print('\n')
print(np.where(b[c]==1)) # 返回指定行的指定元素的位置索引 这里返回了b[c]每行中 值为1的位置索引

观察np.where()的返回值(array([0, 1], dtype=int64), array([4, 7], dtype=int64))可以发现,

对于数组b[c]来说,其中值为1的位置在[0,4]以及[1,7]上,np.where()返回了一个元组,[0]的位置表示行索引,[1]表示列索引。

综上,np.where(b[c]==1)实现的功能是,从b中按c中元素的顺序找到c所指定的行,然后再这些行中发现值为1的位置。

补充:短小精悍算例:Python如何提取数组Numpy array中指定元素的位置索引

数组a的表达式如下:

import numpy as np
a = np.array([1, 2, 3, 2, 2, 3, 0, 8, 3])
print(a)

输出结果:

[1 2 3 2 2 3 0 8 3]

现在要找出a中元素3的位置索引。

r1 = np.where(a==3)
print(type(r1))
print(r1)

输出结果:

<class 'tuple'>

(array([2, 5, 8], dtype=int64),)

可见得到的是一个tuple。

如果要想得到array,采用如下表示:

r1 = np.where(a==3)
print(type(r1[0]))
print(r1[0])

输出结果:

<class 'numpy.ndarray'>

[2 5 8]

还可以使用argwhere()函数进行操作:

r2 = np.argwhere(a==3)
print(type(r2))
print(r2)
print(r2.shape)

输出结果:

<class 'numpy.ndarray'>

[[2]

[5]

[8]]

(3, 1)


以上就是关于“Numpy怎样实现返回指定位置索引”的相关知识,感谢各位的阅读,想要掌握这篇文章的知识点还需要大家自己动手实践使用过才能领会,如果想了解更多相关内容的文章,欢迎关注群英网络,小编每天都会为大家更新不同的知识。
群英智防CDN,智能加速解决方案

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:mmqy2019@163.com进行举报,并提供相关证据,查实之后,将立刻删除涉嫌侵权内容。

猜你喜欢

成为群英会员,开启智能安全云计算之旅

立即注册
专业资深工程师驻守
7X24小时快速响应
一站式无忧技术支持
免费备案服务
免费拨打  400-678-4567
免费拨打  400-678-4567 免费拨打 400-678-4567 或 0668-2555555
在线客服
微信公众号
返回顶部
返回顶部 返回顶部
在线客服
在线客服