NumPy numpy.median Función

Suraj Joshi 30 enero 2023
  1. Sintaxis de numpy.median():
  2. Códigos de ejemplo: numpy.median() Método para encontrar la mediana de un arreglo
  3. Códigos de ejemplo: Poner el parámetro axis en numpy.median() Método para encontrar la mediana de un arreglo a lo largo de un eje particular
  4. Códigos de ejemplo: Establece el parámetro out en numpy.median() Método
  5. Códigos de ejemplo: Establece el parámetro keepdims en numpy.median() Método
NumPy numpy.median Función

La función Numpy numpy.median() calcula la mediana de el array NumPy dada sobre el eje especificado.

Sintaxis de numpy.median():

numpy.median(a, axis=None, out=None, overwrite_input=False, keepdims=False)

Parámetros

a Arreglo u objeto, que podría ser convertido en un arreglo cuya mediana debe ser calculada.
axis encontrar la mediana a lo largo de la fila (axis=0) o la columna (axis=1). Por defecto, la mediana se calcula aplanando el array.
out marcador de posición para el resultado del método np.median().
overwrite_input Booleana. La matriz de entrada será modificada por la llamada al método median() (overwrite_input=True)
keepdims Booleana. Haz que las dimensiones de la salida sean las mismas que las de la entrada (keepdims=True).

Retorna

Arreglo con medianas a lo largo del eje especificado.

Códigos de ejemplo: numpy.median() Método para encontrar la mediana de un arreglo

import numpy as np

a=np.array([[2,3,4],
            [5,6,7],
           [8,9,10]])

median=np.median(a)

print(median)

Producción:

6.0

Calcula la mediana de el array aplanando el array.

Al aplanar el array, nos referimos a colocar todas las filas una tras otra para convertir el array dada en un array de 1-D.

Códigos de ejemplo: Poner el parámetro axis en numpy.median() Método para encontrar la mediana de un arreglo a lo largo de un eje particular

Códigos de ejemplo: numpy.median() Método para encontrar la mediana de un conjunto a lo largo del eje de la columna

Para encontrar la media de el array a lo largo del eje de la columna, establecemos axis=0.

import numpy as np

a=np.array([[2,3],
            [5,6],
           [8,9]])

median=np.median(a,axis=0)

print(median)

Producción:

[5. 6.]

Calcula la mediana de ambas columnas y finalmente devuelve un array con la mediana de cada columna.

Códigos de ejemplo: numpy.median() Método para encontrar la mediana de un arreglo a lo largo del eje de la fila

Para encontrar la mediana de el array a lo largo del eje de la fila, establecemos axis=1.

import numpy as np

a=np.array([[2,3],
            [5,6],
           [8,9]])

median=np.median(a,axis=1)

print(median)

Producción:

[2.5 5.5 8.5]

Calcula la mediana de las tres filas y finalmente devuelve un array con la mediana de cada fila.

Códigos de ejemplo: Establece el parámetro out en numpy.median() Método

import numpy as np

a = np.array([[2, 3], [5, 6], [8, 9]])

median = np.zeros(np.median(a, axis=1).shape)
print(f"median before calculation: {median}")

np.median(a, axis=1, out=median)
print(f"median after calculation: {median}")

Producción:

[2.5 5.5 8.5]

Guarda el resultado del método en la variable median.

Debemos asegurarnos de que la dimensión de la variable a la que se va a asignar la salida sea del mismo tamaño que la de la salida.

Códigos de ejemplo: Establece el parámetro keepdims en numpy.median() Método

import numpy as np

a = np.array([[2, 3], [5, 6], [8, 9]])

print(f"Dimension of Input Array: {median.ndim}")

median = np.median(a, axis=1)
print(f"Dimension of median with 'keepdims=False': {median.ndim}")

median = np.median(a, axis=1, keepdims=True)
print(f"Dimension of median with 'keepdims=True': {median.ndim}")

Producción:

Dimension of Input Array: 2
Dimension of median with 'keepdims=False': 1
Dimension of median with 'keepdims=True': 2

Ajustar keepdims=True preserva el número de dimensiones en el array de salida.

Aquí, el array de entrada a es de 2 dimensiones. Si keepdims=False (valor por defecto), las dimensiones de median pueden ser alteradas pero configurando keepdims=True se preserva el número de dimensiones en la salida del método np.median().

Suraj Joshi avatar Suraj Joshi avatar

Suraj Joshi is a backend software engineer at Matrice.ai.

LinkedIn