Vectorizing for loop in Numpy

Question

is there a way to vectorize this?

waveheight=zeros(10000)
for t in range(10000):
    for j in range(N_frequencysteps):
        waveheight[t] = waveheight[t] + (Abs_S_newwave[j] * cos (K[j] * x - (omega[j] * ((t*0.01) - TimeShift)) + TSi_omega[j] + arg_S_newwave[j]))

eumiro · Accepted Answer · 2011-12-14 13:25:06Z

4

waveheight = (Abs_S_newwave[:,None] * cos(K[:,None] * x - (omega[:,None] * ((arange(10000)[None,:]*0.01) - TimeShift)) + TSi_omega[:,None] + arg_S_newwave[:,None])).sum(axis=0)

This works if all arrays of length N_frequencysteps are 1-D numpy arrays.

answered Dec 14, 2011 at 13:25

eumiro

214k36 gold badges307 silver badges264 bronze badges

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

Comments

silvado · Accepted Answer · 2011-12-14 13:25:10Z

0

At least one step of vectorization would be not to iterate over the elements of waveheight:

waveheight=zeros(10000)
ts = arange(10000)
for j in range(N_frequencysteps):
    waveheight += (Abs_S_newwave[j] * cos(K[j] * x - (omega[j] * 
             ((ts*0.01) - TimeShift)) + TSi_omega[j] + arg_S_newwave[j]))

This assumes that all other variables are scalars.

answered Dec 14, 2011 at 13:25

silvado

18.3k2 gold badges34 silver badges48 bronze badges

Collectives™ on Stack Overflow

Vectorizing for loop in Numpy

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related