Cálculo da Média de Arrays NumPy

Introdução

NumPy é um pacote Python para computação científica que fornece um objeto array de alto desempenho, que é o bloco fundamental para operações matemáticas. A média pode ser facilmente calculada somando todos os itens de um array e dividindo-os pelo número total de elementos do array. A função numpy.mean() na biblioteca NumPy é usada para calcular a média aritmética ao longo do eixo especificado de um array NumPy. Por padrão, a média é calculada sobre o array achatado, a menos que o usuário especifique um eixo.

Dicas para a VM

Após a inicialização da VM, clique no canto superior esquerdo para mudar para a aba Notebook e acessar o Jupyter Notebook para praticar.

Às vezes, pode ser necessário aguardar alguns segundos para que o Jupyter Notebook termine de carregar. A validação das operações não pode ser automatizada devido a limitações no Jupyter Notebook.

Se você enfrentar problemas durante o aprendizado, sinta-se à vontade para perguntar ao Labby. Forneça feedback após a sessão, e resolveremos o problema prontamente para você.

Importar a biblioteca NumPy

O primeiro passo é importar a biblioteca NumPy.

import numpy as np

Criar um array unidimensional

Crie um array unidimensional x com os valores [80, 23, 17, 1, 39].

x = np.array([80, 23, 17, 1, 39])

Calcular a média do array

Use a função numpy.mean() para calcular a média do array unidimensional x.

array_mean = np.mean(x)
print("A média do array de entrada é: ", array_mean)

Criar um array bidimensional

Crie um array bidimensional p com os valores [[14, 19, 12, 34, 43], [16, 8, 28, 8, 20], [25, 5, 55, 1, 2]].

p = np.array([[14, 19, 12, 34, 43], [16, 8, 28, 8, 20], [25, 5, 55, 1, 2]])

Calcular a média do array achatado

Use a função numpy.mean() para calcular a média do array p achatado.

mean_flattened = np.mean(p)
print("A média do array quando axis = None : ", mean_flattened)

Calcular a média ao longo do eixo 0

Use a função numpy.mean() para calcular a média do array p ao longo do eixo 0.

mean_axis_0 = np.mean(p, axis = 0)
print("A média do array quando axis = 0 : ", mean_axis_0)

Calcular a média ao longo do eixo 1

Use a função numpy.mean() para calcular a média do array p ao longo do eixo 1.

mean_axis_1 = np.mean(p, axis = 1)
print("A média do array quando axis = 1 : ", mean_axis_1)

Parâmetro 'Out'

Use a função numpy.mean() com o parâmetro out para colocar o resultado em um array alternativo.

out_arr = np.arange(3)
print("out_arr : ", out_arr)
print("Média de arr, axis = 1: ", np.mean(p, axis = 1, out = out_arr))

Resumo

Neste tutorial, cobrimos a função numpy.mean() da biblioteca NumPy. Explicamos o que é a média, a sintaxe da função mean() e seus parâmetros. Também fornecemos exemplos passo a passo de como usar esta função em arrays unidimensionais e bidimensionais.