Python NumPy实现数组搜索示例详解

 更新时间:2023年05月20日 08:32:41   作者:魔王不会哭  
NumPy是一个开源的Python科学计算库,使用NumPy可以很自然地使用数组和矩阵,这篇文章主要介绍了使用NumPy实现数组搜索,文中通过示例代码介绍的非常详细,需要的朋友们下面随着小编来一起学习吧

搜索数组

可以在数组中搜索(检索)某个值,然后返回获得匹配的索引。

要搜索数组,请使用 where() 方法。

实例

查找值为 4 的索引:

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 4, 4])
x = np.where(arr == 4)
print(x)

运行实例

上例会返回一个元组:(array([3, 5, 6],)

意思就是值 4 出现在索引 3、5 和 6。

实例

查找值为偶数的索引:

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8])
x = np.where(arr%2 == 0)
print(x)

运行实例

实例

查找值为奇数的索引:

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8])
x = np.where(arr%2 == 1)
print(x)

运行实例

搜索排序

有一个名为 searchsorted() 的方法,

该方法在数组中执行二进制搜索,并返回将在其中插入指定值以维持搜索顺序的索引。

假定 searchsorted() 方法用于排序数组。

实例

查找应在其中插入值 7 的索引:

import numpy as np
arr = np.array([6, 7, 8, 9])
x = np.searchsorted(arr, 7)
print(x)

运行实例

例子解释: 应该在索引 1 上插入数字 7,以保持排序顺序。

该方法从左侧开始搜索,并返回第一个索引,其中数字 7 不再大于下一个值。

从右侧搜索

默认情况下,返回最左边的索引,但是我们可以给定 side=‘right’,以返回最右边的索引。

实例

从右边开始查找应该插入值 7 的索引:

import numpy as np
arr = np.array([6, 7, 8, 9])
x = np.searchsorted(arr, 7, side='right')
print(x)

运行实例

例子解释:应该在索引 2 上插入数字 7,以保持排序顺序。

该方法从右边开始搜索,并返回第一个索引,其中数字 7 不再小于下一个值。

多个值

要搜索多个值,请使用拥有指定值的数组。

实例

查找应在其中插入值 2、4 和 6 的索引:

import numpy as np
arr = np.array([1, 3, 5, 7])
x = np.searchsorted(arr, [2, 4, 6])
print(x)

运行实例

返回值是一个数组:[1 2 3] 包含三个索引,其中将在原始数组中插入 2、4、6 以维持顺序。

到此这篇关于Python NumPy实现数组搜索示例详解的文章就介绍到这了,更多相关Python NumPy数组搜索内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • python使用列表的最佳方案

    python使用列表的最佳方案

    这篇文章主要介绍了python使用列表的最佳方式,帮助大家更好的理解和学习python,感兴趣的朋友可以了解下
    2020-08-08
  • python实现字符串完美拆分split()的方法

    python实现字符串完美拆分split()的方法

    今天小编就为大家分享一篇python实现字符串完美拆分split()的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • PyCharm导入numpy库的几种方式

    PyCharm导入numpy库的几种方式

    今天给大家带来的是关于Python的相关知识,文章围绕着PyCharm导入numpy库的几种方式展开,文中有非常详细的解释及代码示例,需要的朋友可以参考下
    2021-06-06
  • python实现PyEMD经验模态分解残差量分析

    python实现PyEMD经验模态分解残差量分析

    这篇文章主要为大家介绍了PyEMD经验模态分解及变体残余量分析,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-05-05
  • Python 动态绑定属性和方法 

    Python 动态绑定属性和方法 

    这篇文章主要介绍了Python 动态绑定属性和方法, 动态语言与静态语言有很多不同,最大的特性之一就是可以实现动态的对类和实例进行修改,在Python中,我们创建了一个类后可以对实例和类绑定心的方法或者属性,实现动态绑定,下面详细介绍需要的朋友可以参考一下
    2022-03-03
  • 详解Python遍历字典的键和值

    详解Python遍历字典的键和值

    这篇文章主要通过一些简单的示例为大家介绍一下Python中遍历字典的键和值的相关知识,文中的示例代码讲解详细,感兴趣的小伙伴可以了解一下
    2023-03-03
  • python 实现百度网盘非会员上传超过500个文件的方法

    python 实现百度网盘非会员上传超过500个文件的方法

    这篇文章主要介绍了python 实现百度网盘非会员上传超过500个文件的方法,帮助大家更好的利用python解决问题,感兴趣的朋友可以了解下
    2021-01-01
  • 安装python-docx后,无法在pycharm中导入的解决方案

    安装python-docx后,无法在pycharm中导入的解决方案

    这篇文章主要介绍了安装python-docx后,无法在pycharm中导入的解决方案,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-03-03
  • python网络爬虫之如何伪装逃过反爬虫程序的方法

    python网络爬虫之如何伪装逃过反爬虫程序的方法

    本篇文章主要介绍了python网络爬虫之如何伪装逃过反爬虫程序的方法,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-11-11
  • 使用Python和Plotly绘制各种类型3D图形的方法

    使用Python和Plotly绘制各种类型3D图形的方法

    Python语言拥有丰富的数据可视化库,其中Plotly是一款流行的工具,提供了绘制高质量三维图形的功能,本文将介绍如何使用Python和Plotly来绘制各种类型的3D图形,并给出代码实例,需要的朋友可以参考下
    2024-05-05

最新评论