Dominando el cálculo de la media de matrices NumPy

Introducción

NumPy es un paquete de Python para el cálculo científico que proporciona un objeto de matriz (array) de alto rendimiento, que es el componente fundamental para las operaciones matemáticas. La media se puede calcular fácilmente sumando todos los elementos de una matriz y dividiéndolos por el número total de elementos de la matriz. La función numpy.mean() en la biblioteca NumPy se utiliza para calcular la media aritmética a lo largo del eje especificado de una matriz de NumPy. Por defecto, la media se calcula sobre la matriz aplanada, a menos que el usuario especifique un eje.

Consejos para la MV

Después de que la máquina virtual (VM) haya terminado de iniciarse, haga clic en la esquina superior izquierda para cambiar a la pestaña Notebook y acceder a Jupyter Notebook para practicar.

A veces, es posible que deba esperar unos segundos para que Jupyter Notebook termine de cargarse. La validación de las operaciones no se puede automatizar debido a las limitaciones de Jupyter Notebook.

Si tiene problemas durante el aprendizaje, no dude en preguntar a Labby. Proporcione comentarios después de la sesión y resolveremos rápidamente el problema para usted.

Importar la biblioteca NumPy

El primer paso es importar la biblioteca NumPy.

import numpy as np

Crear una matriz unidimensional

Crea una matriz unidimensional x con los valores [80, 23, 17, 1, 39].

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

Calcular la media de la matriz

Utiliza la función numpy.mean() para calcular la media de la matriz unidimensional x.

array_mean = np.mean(x)
print("The mean of the input array is: ", array_mean)

Crear una matriz bidimensional

Crea una matriz bidimensional p con los 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 la media de la matriz aplanada

Utiliza la función numpy.mean() para calcular la media de la matriz p aplanada.

mean_flattened = np.mean(p)
print("The mean of the array when axis = None : ", mean_flattened)

Calcular la media a lo largo del eje 0

Utiliza la función numpy.mean() para calcular la media de la matriz p a lo largo del eje 0.

mean_axis_0 = np.mean(p, axis = 0)
print("The mean of the array when axis = 0 : ", mean_axis_0)

Calcular la media a lo largo del eje 1

Utiliza la función numpy.mean() para calcular la media de la matriz p a lo largo del eje 1.

mean_axis_1 = np.mean(p, axis = 1)
print("The mean of the array when axis = 1 : ", mean_axis_1)

Parámetro de salida

Utiliza la función numpy.mean() con el parámetro out para colocar el resultado en una matriz alternativa.

out_arr = np.arange(3)
print("out_arr : ", out_arr)
print("Mean of arr, axis = 1: ", np.mean(p, axis = 1, out = out_arr))

Resumen

En este tutorial, cubrimos la función numpy.mean() de la biblioteca NumPy. Explicamos qué es la media, la sintaxis de la función mean() y sus parámetros. También proporcionamos ejemplos paso a paso de cómo usar esta función en matrices unidimensionales y bidimensionales.

Cálculo de la media de matrices NumPy