numpy.searchsorted() en Python

numpy.searchsorted() La función se usa para encontrar los índices en una array ordenada de tal manera que, si los elementos se insertan antes de los índices, el orden de arr aún se conservaría. Aquí, la búsqueda binaria se usa para encontrar los índices de inserción requeridos.

Sintaxis: numpy.searchsorted(arr, num, side=’left’, sorter=Ninguno)

Parámetros:
arr: [array_like] Array de entrada. Si sorter es None, entonces debe ordenarse en orden ascendente; de ​​lo contrario, sorter debe ser una array de índices que lo ordenen.
num: [array_like] Los valores que queremos insertar en arr.
lado: [‘izquierda’, ‘derecha’], opcional. Si es ‘izquierda’, se proporciona el índice de la primera ubicación adecuada encontrada. Si es ‘correcto’, devuelve el último índice de este tipo. Si no hay un índice adecuado, devuelve 0 o N (donde N es la longitud de a).
num : [array_like, Optional] array de índices enteros que ordenan la array a en orden ascendente. Por lo general, son el resultado de argsort.

Devuelve: [índices], array de puntos de inserción con la misma forma que num.

Código #1: Trabajando

# Python program explaining
# searchsorted() function
   
import numpy as geek
  
# input array
in_arr = [2, 3, 4, 5, 6]
print ("Input array : ", in_arr)
  
# the number which we want to insert
num = 4
print("The number which we want to insert : ", num) 
    
out_ind = geek.searchsorted(in_arr, num) 
print ("Output indices to maintain sorted array : ", out_ind)

Producción :

Input array :  [2, 3, 4, 5, 6]
The number which we want to insert :  4
Output indices to maintain sorted array :  2

 
Código #2:

# Python program explaining
# searchsorted() function
   
import numpy as geek
  
# input array
in_arr = [2, 3, 4, 5, 6]
print ("Input array : ", in_arr)
  
# the number which we want to insert
num = 4
print("The number which we want to insert : ", num)   
  
out_ind = geek.searchsorted(in_arr, num, side ='right') 
print ("Output indices to maintain sorted array : ", out_ind)

Producción :

Input array :  [2, 3, 4, 5, 6]
The number which we want to insert :  4
Output indices to maintain sorted array :  3

 
Código #3:

# Python program explaining
# searchsorted() function
   
import numpy as geek
  
# input array
in_arr = [2, 3, 4, 5, 6]
print ("Input array : ", in_arr)
  
# the numbers which we want to insert
num = [4, 8, 0]
print("The number which we want to insert : ", num)   
  
out_ind = geek.searchsorted(in_arr, num) 
print ("Output indices to maintain sorted array : ", out_ind)

Producción :

Input array :  [2, 3, 4, 5, 6]
The number which we want to insert :  [4, 8, 0]
Output indices to maintain sorted array :  [2 5 0]