How to get rid of for loop in my function?

Asked Sep 02 '21 at 08:54

Active Sep 02 '21 at 10:46

Viewed 49 times

y = numpy.zeros(len(data))
for i in range(len(data)):
    if i == 0:
        y[i] = (1.0-dec) * data[i]
    else:
        y[i] = (1.0-dec) * data[i] + (dec*y[i-1])

y

So for example I have

data = np.array([100,200,300,400,500])
dec = 0.1

output:

array([ 90.   , 189.   , 288.9  , 388.89 , 488.889])

But for a huge dataset of data the "for" loop will take a lot of time to execute, so I was seeking help from someone that can we optimize it using some function from numpy.

edited Sep 02 '21 at 10:46

asked Sep 02 '21 at 08:54

Akilesh

please explain what you mean by "getting rid" of the for loop? maybe explain what the code is supposed to do and why you used a for loop in the first place? – Almog-at-Nailo Sep 02 '21 at 09:04
1

I think they mean how to vectorize it, but it would help to know what it's supposed to do first. – Iguananaut Sep 02 '21 at 09:11
@Iguananaut yeah I mean how to vectorize. – Akilesh Sep 02 '21 at 09:16
2

you can use [numba](http://numba.pydata.org/) to optimize run time as it is. – yann ziselman Sep 02 '21 at 09:20
@Iguananaut still don't understand it, can you write it down in my case? – Akilesh Sep 02 '21 at 09:48

How to get rid of for loop in my function?

0 Answers0