在python中的numpy中查找范围内的元素

有时在使用numpy库处理数据时，我们可能需要过滤掉特定范围内的某些数字。这可以通过使用一些numpy中可用的内置方法来实现。

与和运算符

在这种方法中，我们采用一个numpy数组，然后对其应用逻辑_和函数。numpy中的where子句也用于应用and条件。结果是一个数组，显示满足所需范围条件的元素的位置。

import numpy as np

A = np.array([5, 9, 11, 4, 31, 27,8])

# printing initial array
print("Given Array : ", A)

# Range 6 to 15
res = np.where(np.logical_and(A >= 6, A <= 15))

# Result
print("Array with condition : ", res)

输出结果

运行上面的代码给我们以下结果-

Given Array : [ 5 9 11 4 31 27 8]
Array with condition : (array([1, 2, 6], dtype=int32),)

使用*

在这种方法中，我们使用*运算符。它给出结果作为实际值，而不是值在数组中的位置。

import numpy as np

A = np.array([5, 9, 11, 4, 31, 27,8])

# printing initial array
print("Given Array : ", A)

# Range 6 to 15
res = A [ (A >=6) * (A <= 15)]

# Result
print("Array with condition : ", res)

输出结果

运行上面的代码给我们以下结果-

Given Array : [ 5 9 11 4 31 27 8]
Array with condition : [ 9 11 8]