Averaging over every n elements of a numpy array

python numpy average
user1654183 picture user1654183 · Apr 11, 2013 · Viewed 33.3k times · Source

I have a numpy array. I want to create a new array which is the average over every consecutive triplet of elements. So the new array will be a third of the size as the original.

As an example:

 np.array([1,2,3,1,2,3,1,2,3])

should return the array:

 np.array([2,2,2])

Can anyone suggest an efficient way of doing this? I'm drawing blanks.

Answer

Jaime picture Jaime · Apr 11, 2013

If your array arr has a length divisible by 3:

np.mean(arr.reshape(-1, 3), axis=1)

Reshaping to a higher dimensional array and then performing some form of reduce operation on one of the additional dimensions is a staple of numpy programming.

Related questions

np.mean() vs np.average() in Python NumPy?

I notice that In [30]: np.mean([1, 2, 3]) Out[30]: 2.0 In [31]: np.average([1, 2, 3]) Out[31]: 2.0 However, there should be some differences, since after all they are two different functions. What are the differences between them?

Read More
Weighted average using numpy.average

I have an array: In [37]: bias_2e13 # our array Out[37]: [1.7277990734072355, 1.9718263893212737, 2.469657573252167, 2.869022991373125, 3.314720313010104, 4.232269039271717] The error on each value in the array is: In [38]: bias_error_2e13 # the error on each value Out[38]: array([ 0.13271387, 0.06842465, 0.06937965, 0.23886647, 0.30458249, 0.57906816]) Now I divide the error on each value by 2: In [39]: …

Read More
How to convert a list of strings into a numeric numpy array?

I want to be able to calculate the mean, min and max of A: import numpy as np A = ['33.33', '33.33', '33.33', '33.37'] NA = np.asarray(A) AVG = np.mean(NA, axis=0) print AVG This does …

Read More