numpy.partition() en Python

numpy.partition() La función se usa para crear una copia particionada de la array de entrada con sus elementos reorganizados de tal manera que el valor del elemento en la k-ésima posición está en la posición en la que estaría en una array ordenada. Todos los elementos menores que el k-ésimo elemento se mueven antes de este elemento y todos los iguales o mayores se mueven detrás de él. El orden de los elementos en las dos particiones no está definido.

Sintaxis: numpy.partition(arr, kth, axis=-1, kind=’introselect’, order=Ninguno)

Parámetros:
arr: [array_like] Array de entrada.
kth : [int o secuencia de enteros] Índice del elemento por el que se va a particionar.
axis : [int o None] Eje a lo largo del cual ordenar. Si es Ninguno, la array se aplana antes de ordenar. El valor predeterminado es -1, que ordena a lo largo del último eje.
kind : Algoritmo de selección. El valor predeterminado es ‘introselect’.
orden: [str o lista de str] Cuando arr es una array con campos definidos, este argumento especifica qué campos comparar primero, segundo, etc.

Retorno: [ndarray] Array particionado del mismo tipo y forma que arr.

Código #1:

# Python program explaining
# partition() function
   
import numpy as geek
  
# input array
in_arr = geek.array([ 2, 0,  1, 5, 4, 9])
print ("Input array : ", in_arr) 
  
out_arr = geek.partition(in_arr, 3)
print ("Output partitioned array : ", out_arr)

Input array :  [2 0 1 5 4 9]
Output partitioned array :  [0 1 2 4 5 9]

 
Código #2:

# Python program explaining
# partition() function
   
import numpy as geek
  
# input array
in_arr = geek.array([ 2, 0,  1, 5, 4, 9, 3])
print ("Input array : ", in_arr) 
  
out_arr = geek.partition(in_arr, (0, 3))
print ("Output partitioned array : ", out_arr)

Input array :  [2 0 1 5 4 9 3]
Output partitioned array :  [0 1 2 3 4 9 5]