python numpy ndarray element-wise mean

Aerin picture Aerin · May 25, 2016 · Viewed 22.7k times · Source

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?

Answer

aldanor picture aldanor · May 25, 2016

You can just use np.mean directly:

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