Search code examples
pythonnumpyvectorization

How to get the element-wise mean of an ndarray

I'd like to calculate element-wise average of numpy ndarray.

In [56]: a = np.array([10, 20, 30])

In [57]: b = np.array([30, 20, 20])

In [58]: c = np.array([50, 20, 40])

What I want:

[30, 20, 30]

Is there any in-built function for this operation, other than vectorized sum and dividing?

Solution

  • You can just use np.mean directly:

    >>> np.mean([a, b, c], axis=0)
array([ 30.,  20.,  30.])