Write function for exponential aggregation using R/Python

Question

Input -->[10,20,30,40,50]  
Desired Output  ---> [10,19.21,56.51,110.83,181.14] 
 
**Calculation**  
10-->10  
20-->20*exp(-0.04)  
30-->30*exp(-0.04)+((30*exp(-0.04))*exp(-0.04))  
40--->40*exp(-0.04)+((40*exp(-0.04))*exp(-0.04))+(((40*exp(-0.04))*exp(-0.04))*exp(-0.04)))

Attaching calculation table for easy understanding

Please help me to write function to resolve the above issue using R and python code

Answer 1

Not sure how to do this in R but it is pretty straightforward in Python. You will need numpy for this in order to get the exponential of -0.04.

import numpy as np

def data_agg(data):
    results = []
    exp = np.exp(-0.04)
    for num in data:
        iters = (num // 10) - 1
        if iters == 0:
            results.append(num)
        else:
            tmp = 0
            while iters > 0:
                tmp += num * pow(exp, iters)
                iters -= 1
            results.append(round(tmp, 3))
    return results

data_agg(data)

>> [10, 19.216, 56.517, 110.833, 181.149]