mean along axis numpy code example

Example 1: np.array average row

>>> a = np.array([[1, 2], [3, 4]])
>>> np.mean(a)
2.5
>>> np.mean(a, axis=0)
array([2., 3.])
>>> np.mean(a, axis=1)
array([1.5, 3.5])

Example 2: numpy mean

>> import numpy as np 
>> a=[1,2,3,4,5]
>> np.mean(a)
3.0

Example 3: python mean ndarray

# define ndarray a
a = np.array([[1, 2], [3, 4]])
# get the mean 
np.mean(a, axis=None)