Numpy中的数组搜索:np.where方法详细介绍

np.where和np.searchsorted同属于Numpy数组搜索的一部分,这里先介绍简单的where

import numpy as np
a = np.array([1, 2, 3, 4, 5])
b = np.where(a == 5)

print(b)

where方法将会返回一个元祖

(array([4]),)

此外还将介绍一个搜索奇数和偶数的方法(数组全都默认使用最上面的a数组)
可见,简单的判断余数即可

c = np.where(a%2 == 0)
print(c)

d = np.where(a%2 == 1)
print(d)

返回:

(array([1, 3]),)
(array([0, 2, 4]),)

关于np.where方法到这里就结束啦

如果想要了解更多关于Numpy的知识,欢迎来关注我哦,我会在后续用清晰明了的语言,讲述Numpy的各个基础知识。
也欢迎关注微信公众号:MathematiCS Studio,我将会定期在里面写有趣的数学和计算机知识与你分享,中英双语哦~

你可能感兴趣的:(Numpy,python,数据分析,numpy)