mean of elements i and i+1 in a numpy array

Question

Out of curiosity, is there a specific numpy function to do the following (which would supposedly be faster):

a = np.array((0,2,4))
b = np.zeros(len(a) - 1)
for i in range(len(b)):
    b[i] = a[i:i+2].mean()

print(b)
#prints [1,3]

Cheers

Sven Marnach · Accepted Answer · 2011-05-05 14:14:16Z

7

You could use

b = (a[1:] + a[:-1]) / 2.

to avoid the Python loop.

answered May 5, 2011 at 14:14

Sven Marnach

608k123 gold badges968 silver badges865 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Neat, I like it :-) I guess I have not worked with numpy for too long! Thank you!