numpy 实现返回指定行的指定元素的位置索引
更新时间：2021年05月10日 09:46:46 作者：徐亦快
这篇文章主要介绍了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)
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。
相关文章
在本篇文章里小编给大家整理了关于Python生成pdf目录书签的实例方法，有需要的朋友们可以学习下。2020-10-10
Python使用textract实现从各种文件中提取文本信息
textract是一个强大的Python库,可以用于从各种文件格式中提取文本,本文将介绍textract的使用场景,以及一些常用的Python代码案例,希望对大家有所帮助2024-01-01
这篇文章主要介绍了Django框架之django admin的命令行知识,文中通过示例代码给大家介绍的非常详细，需要的朋友可以参考下2021-09-09
这篇文章主要介绍了python 的numpy库中的mean()函数用法介绍，具有很好对参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-03-03
这篇文章主要介绍了Python日志logging模块功能与用法,结合实例形式详细分析了Python日志logging模块的基本功能、原理、用法及操作注意事项,需要的朋友可以参考下2020-04-04
今天小编就为大家分享一篇Tensorflow分批量读取数据教程，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
利用Python-iGraph如何绘制贴吧/微博的好友关系图详解
这篇文章主要给大家介绍了关于利用Python-iGraph如何绘制贴吧/微博好友关系图的相关资料，文中显示介绍了在windows系统下安装python-igraph的步骤，然后通过示例代码演示了绘制好友关系图的方法，需要的朋友可以参考下。2017-11-11
在本篇文章里小编给大家整理了一篇关于python匿名函数的实例用法，有兴趣的朋友们可以学习参考下。2021-03-03
FFmpeg是业内有名的开源图像视频处理程序,在许多视频剪辑软件、图像处理软件中,都使用的FFmpeg,还有比如OBS这样的导播软件里面也使用了FFmpeg,FFmpeg的功能十分强大,远不止视频压缩的功能,本文介绍使用FFmpeg来无损压缩视频文件的操作方法,感兴趣的朋友一起看看吧2023-12-12
在本篇文章里小编给大家整理了关于python程序编译相关的知识点内容，有兴趣的朋友们参考学习下。2020-06-06
最新评论