Resumo : neste tutorial, você aprenderá como usar a mean()
função numpy para calcular a média dos elementos de um array.
Introdução à função NumPy mean()
A
função retorna a média dos elementos de um array. Aqui está a sintaxe da mean()
função:mean()
numpy.mean(a, axis=None, dtype=None, out=None, keepdims=<no value>, *, where=<no value>)
Linguagem de código: Python ( python )
Nesta sintaxe:
a
é um array que você deseja calcular a média dos elementos.axis
é o eixo, se especificado, retornará a média dos elementos nesse eixo.
Para entender mais sobre outros parâmetros e seus usos, verifique a documentação da função numpy mean() .
Exemplos de funções NumPy mean()
Vejamos alguns exemplos de uso da mean()
função.
1) Usando a função NumPy mean() no exemplo de array 1-D
O exemplo a seguir usa a mean()
função para calcular a média dos números em uma matriz:
import numpy as np
a = np.array([1, 2, 3])
average = np.mean(a)
print(average)
Linguagem de código: Python ( python )
Saída:
2.0
Linguagem de código: Python ( python )
Como funciona.
Primeiro, crie um array com três números:
a = np.array([1, 2, 3])
Linguagem de código: Python ( python )
Segundo, calcule a média dos elementos do array a
usando a mean()
função:
average = np.mean(a)
Linguagem de código: Python ( python )
Terceiro, exiba a média:
print(average)
Linguagem de código: Python ( python )
A saída é 2,0 porque (1 + 2 + 3) / 3 = 2,0
2) Usando a função NumPy mean() no exemplo de array 2-D
O exemplo a seguir usa a mean()
função para calcular a média dos elementos no eixo 0:
import numpy as np
a = np.array([
[1, 2, 3],
[4, 5, 6]
])
average = np.mean(a, axis=0)
print(average)
Linguagem de código: Python ( python )
Saída:
[2.5 3.5 4.5]
Linguagem de código: Python ( python )
Resumo
- Use a função numpy
mean()
para calcular a média dos elementos em uma matriz.