NumPy numpy.median Função

Suraj Joshi 30 janeiro 2023
  1. Sintaxe de numpy.median():
  2. Códigos de exemplo: numpy.median() Método para encontrar a mediana de um Array
  3. Códigos de exemplo: Definir axis Parâmetro em numpy.median() Método para encontrar a mediana de um array ao longo de um eixo em particular
  4. Códigos de exemplo: Definir out Parâmetro em numpy.median() Método
  5. Códigos de exemplo: Definir keepdims Parâmetro em numpy.median() Método
NumPy numpy.median Função

Python Numpynumpy.median() calcula a mediana de uma dado array NumPy sobre o eixo especificado.

Sintaxe de numpy.median():

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

Parâmetros

a Array ou Objecto, que poderia ser convertido em um array cuja mediana deve ser calculada.
axis encontrar mediana ao longo da linha (axis=0) ou coluna (axis=1). Por defeito, a mediana é calculada achatando a matriz.
out espaço reservado para o resultado do método np.median()
overwrite_input Booleano. A matriz de entrada será modificada pela chamada para o método median() (overwrite_input=True)
keepdims Booleano. Tornar as dimensões da saída iguais às da entrada (keepdims=True).

Devolver

Array com medianas ao longo do eixo especificado.

Códigos de exemplo: numpy.median() Método para encontrar a mediana de um Array

import numpy as np

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

median=np.median(a)

print(median)

Resultado:

6.0

Calcula a mediana do array aplanando a matriz.

Ao aplanar a matriz, referimo-nos a colocar todas as filas uma após a outra para converter a matriz dada em 1-D.

Códigos de exemplo: Definir axis Parâmetro em numpy.median() Método para encontrar a mediana de um array ao longo de um eixo em particular

Códigos de exemplo: numpy.median() Método para encontrar a mediana de um array ao longo de um eixo de coluna

Para encontrar a média do array ao longo do eixo da coluna, definimos axis=0.

import numpy as np

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

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

print(median)

Resultado:

[5. 6.]

Calcula a mediana de ambas as colunas e finalmente devolve um array com a mediana de cada coluna.

Códigos de exemplo: numpy.median() Método para encontrar a mediana de um array ao longo do eixo da coluna

Para encontrar a mediana do array ao longo do eixo da linha, definimos axis=1.

import numpy as np

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

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

print(median)

Resultado:

[2.5 5.5 8.5]

Calcula a mediana para as três filas e finalmente devolve um array com a mediana de cada fila.

Códigos de exemplo: Definir out Parâmetro em 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}")

Resultado:

[2.5 5.5 8.5]

Guarda o resultado do método na variável median.

Temos de assegurar que a dimensão da variável à qual a saída deve ser atribuída é da mesma dimensão que a da saída.

Códigos de exemplo: Definir keepdims Parâmetro em 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}")

Resultado:

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

A definição keepdims=True preserva o número de dimensões no array de saída.

Aqui, a matriz de entrada a é bidimensional. Se keepdims=False (valor padrão), as dimensões de median podem ser alteradas, mas a definição de keepdims=True preserva o número de dimensões na saída do método np.median().

Suraj Joshi avatar Suraj Joshi avatar

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

LinkedIn