Encontrar los k valores más pequeños de una array NumPy

En este artículo, veamos cómo encontrar el número k de los valores más pequeños de una array NumPy.

Ejemplos:

Input: [1,3,5,2,4,6] 
k = 3

Output: [1,2,3] 

Método 1: Usar np.sort()

Acercarse:

  1. Cree una array NumPy.
  2. Determine el valor de k.
  3. Ordene la array en orden ascendente usando el método sort().
  4. Imprime los primeros valores k de la array ordenada.

Python3

# importing the modules
import numpy as np
  
# creating the array 
arr = np.array([23, 12, 1, 3, 4, 5, 6])
print("The Original Array Content")
print(arr)
  
# value of k
k = 4
  
# sorting the array
arr1 = np.sort(arr)
  
# k smallest number of array
print(k, "smallest elements of the array")
print(arr1[:k])

Producción:

The Original Array Content
[23 12  1  3  4  5  6]
4 smallest elements of the array
[1 3 4 5]

Método 2: Usar np.argpartition() 

Acercarse:

  1. Cree una array NumPy.
  2. Determine el valor de k.
  3. Obtenga los índices de los k elementos más pequeños usando el método argpartition().
  4. Obtenga los primeros valores k de la array obtenida de argpartition() e imprima sus valores de índice con respecto a la array original.

Python3

# importing the module
import numpy as np
  
# creating the array 
arr = np.array([23, 12, 1, 3, 4, 5, 6])
print("The Original Array Content")
print(arr)
  
# value of k
k = 4
  
# using np.argpartition()
result = np.argpartition(arr, k)
  
# k smallest number of array
print(k, "smallest elements of the array")
print(arr[result[:k]])

Producción:

The Original Array Content
[23 12  1  3  4  5  6]
4 smallest elements of the array
[4 3 1 5]

Publicación traducida automáticamente

Artículo escrito por hupphurr y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *