【Python语言基础】——Python NumPy 数组搜索_python numpy 查找

作者：盐析白兔 | 2024-04-04 03:58:17

踩

python numpy 查找

Python语言基础——Python NumPy 数组搜索

文章目录

Python语言基础——Python NumPy 数组搜索
一、Python NumPy 数组搜索

一、Python 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])

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/357253