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数组搜索内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Pyside6开发使用Qt Designer的示例代码

    Pyside6开发使用Qt Designer的示例代码

    本文主要介绍了Pyside6开发使用Qt Designer的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2025-05-05
  • 详解Python中递归函数的原理与使用

    详解Python中递归函数的原理与使用

    如果一个函数,可以自己调用自己,那么这个函数就是一个递归函数。本文将详细讲解Python中递归函数的使用与原理,感兴趣的可以了解一下
    2022-05-05
  • 用python实现一个让人戒不掉的百变款消消乐

    用python实现一个让人戒不掉的百变款消消乐

    消消乐的热门程度几乎赶上王者荣耀,你是否也有收到过好友邀请你帮解锁关卡的时候,今天小编带你用python编写一个自己的消消乐升级版,同学请往下看
    2021-09-09
  • 浅谈Pytorch中的自动求导函数backward()所需参数的含义

    浅谈Pytorch中的自动求导函数backward()所需参数的含义

    今天小编就为大家分享一篇浅谈Pytorch中的自动求导函数backward()所需参数的含义,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-02-02
  • Python服务器创建虚拟环境跑代码

    Python服务器创建虚拟环境跑代码

    本文主要介绍了Python服务器创建虚拟环境跑代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-07-07
  • python web.py服务器与客户端的实现示例

    python web.py服务器与客户端的实现示例

    本文介绍了使用Python的web.py库搭建一个简单的Web服务器,并演示了自定义端口、URL映射以及POST和GET请求的处理,具有一定的参考价值,感兴趣的可以了解一下
    2025-10-10
  • Python强大的语法支持你知道吗

    Python强大的语法支持你知道吗

    这篇文章主要为大家介绍了Python强大的语法,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助,希望能够给你带来帮助
    2021-11-11
  • Python实现多种场景下查找并高亮Word文档中的文本

    Python实现多种场景下查找并高亮Word文档中的文本

    在处理长篇 Word 文档时,快速定位关键信息并将其突出显示是一项非常实用的技能,本文将深入探讨如何使用 Python 实现多种场景下的 Word 文档文本查找与高亮功能,有需要的小伙伴可以了解下
    2026-03-03
  • python基于itchat模块实现微信防撤回

    python基于itchat模块实现微信防撤回

    这篇文章主要为大家详细介绍了python实现微信防撤回,基于itchat模块,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-04-04
  • Python定义空函数的6种方法小结

    Python定义空函数的6种方法小结

    本文主要介绍了Python中定义空函数的6种方法,包括注释、pass、return、Ellipsis等,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2025-01-01

最新评论