Python | Найти элементы в пределах диапазона в numpy

Опубликовано: 12 Апреля, 2022

Given numpy array, the task is to find elements within some specific range. Let’s discuss some ways to do the task.

Method #1: Using np.where()

# python code to demonstrate
# finding elements in range
# in numpy array
import numpy as np
  
ini_array = np.array([1, 2, 3, 45, 4, 7, 810, 9, 6])
  
# printing initial array
print("initial_array : ", str(ini_array));
  
# find elements in range 6 to 10
result = np.where(np.logical_and(ini_array>= 6, ini_array<= 10))
  
# printing result
print("resultant_array : ", result)

Output:

initial_array :  [  1   2   3  45   4   7 810   9   6]
resultant_array :  (array([5, 7, 8]),)

Method #2: Using numpy.searchsorted()

# Python code to demonstrate
# finding elements in range
# in numpy array
  
import numpy as np
  
ini_array = np.array([1, 2, 3, 45, 4, 7, 9, 6])
  
# printing initial array
print("initial_array : ", str(ini_array));
  
  
# find elements in range 6 to 10
start = np.searchsorted(ini_array, 6, "left")
end = np.searchsorted(ini_array, 10, "right")
result = np.arange(start, end)
  
# printing result
print("resultant_array : ", result)

Output:

initial_array :  [ 1  2  3 45  4  7  9  6]
resultant_array :  [5 6 7]

Метод № 3: Использование *

Внимание компьютерщик! Укрепите свои основы с помощью базового курса программирования Python и изучите основы.

Для начала подготовьтесь к собеседованию. Расширьте свои концепции структур данных с помощью курса Python DS. А чтобы начать свое путешествие по машинному обучению, присоединяйтесь к курсу Машинное обучение - базовый уровень.

Python

Python | Найти элементы в пределах диапазона в numpy

РЕКОМЕНДУЕМЫЕ СТАТЬИ