Averaging numpy array over for loop?

Question

I am calculating a numpy array each iteration of a for loop. How do I average that?

For example:

for i in range(5):
    array_that_I_calculate = some_calculation(x,y,z)

ohk, just append them to a list and then use np.average over axis 0 of that list of arrays. details in my answer — Akshay Sehgal
– Akshay Sehgal, Commented Jan 9, 2023 at 11:43

Akshay Sehgal · Accepted Answer · 2023-01-09 11:40:51Z

1

Try this -

Append the array_that_I_calculate at each iteration into a list_of_arrays
After the loop ends, take np.average() of list_of_arrays over axis=0

import numpy as np


##### IGNORE #####
#dummy function that returns (2000,1) array

def some_calculation(x=None,y=None,z=None)
    return np.random.random((2000,1))


##### SOLUTION #####

list_of_arrays = []                                  #<-----

for i in range(5):
    array_that_I_calculate = some_calculation(x,y,z)
    list_of_arrays.append(array_that_I_calculate)    #<-----
    
averaged_array = np.average(list_of_arrays, axis=0)  #<-----
print(averaged_array.shape)

(2000,1)

answered Jan 9, 2023 at 11:40

Akshay Sehgal

19.4k3 gold badges26 silver badges57 bronze badges

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

1 Comment

Akshay Sehgal Over a year ago

do let me know if that solved your question. feel free to mark the answer if it did. thanks!

Collectives™ on Stack Overflow

Averaging numpy array over for loop?

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related