Calcule o Modo de array no NumPy

1. Calcule o modo de um array NumPy com a função scipy.stats.mode()
2. Calcule o modo de um array NumPy com a função numpy.unique()

Este tutorial irá apresentar como calcular o modo de um array NumPy em Python.

Calcule o modo de um array NumPy com a função scipy.stats.mode()

O modo é o valor mais repetido em uma coleção. A biblioteca scipy.stats contém muitas funções relacionadas às estatísticas. A função mode() dentro da biblioteca scipy.stats encontra o modo de um array em Python. Ele recebe um array como argumento de entrada e retorna um array dos valores mais comuns dentro do array de entrada. Para que este método funcione, temos que instalar o pacote scipy. O comando para instalá-lo é fornecido abaixo.

pip install scipy

O exemplo de código a seguir nos mostra como calcular o modo dentro de um array NumPy com a função scipy.stats.mode().

import numpy as np
from scipy import stats

array = np.array([1, 2, 3, 4, 4, 5])
mode = stats.mode(array)
print(mode[0])

Resultado:

[4]

Primeiro criamos o array array com a função np.array(). Em seguida, calculamos o modo com a função scipy.stats.mode() e armazenamos o resultado dentro do array mode. No final, exibimos o valor mais repetido, imprimindo o primeiro elemento do array mode.

Calcule o modo de um array NumPy com a função numpy.unique()

Se quisermos usar o pacote NumPy apenas para encontrar o modo, podemos usar a função numpy.unique(). A função numpy.unique() recebe um array como argumento de entrada e retorna um array de todos os elementos únicos dentro do array de entrada. Também podemos especificar o parâmetro return_count como True para obter o número de vezes que cada elemento exclusivo é repetido dentro do array de entrada.

import numpy as np

array = np.array([1, 2, 3, 4, 4, 5])
vals, counts = np.unique(array, return_counts=True)
index = np.argmax(counts)
print(vals[index])

Resultado:

4

No código acima, calculamos o modo do array NumPy array com as funções np.unique() e np.argmax() em Python. Primeiro criamos o array array com a função np.array(). Em seguida, usamos a função np.unique() e armazenamos os valores exclusivos dentro do array vals e o número de vezes que cada valor é repetido dentro do array counts. Em seguida, calculamos o valor máximo dentro do array counts com a função np.argmax() e armazenamos o valor dentro da variável index. No final, exibimos o modo imprimindo o valor no índice index do array vals.