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:
- Cree una array NumPy.
- Determine el valor de k.
- Ordene la array en orden ascendente usando el método sort().
- 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:
- Cree una array NumPy.
- Determine el valor de k.
- Obtenga los índices de los k elementos más pequeños usando el método argpartition().
- 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]