python - 如何使用 numpy.repeat 构建一个数组，其中每个元素比前一个元素多 1%？

Question

我的初始值为salary=500。我希望它每年增加 1%，这样我的数组就会像 [500,550,605,665.5] 等。本质上是将 1.1 乘以每个先前的值。

我为此使用了一个循环，但很好奇是否可以使用numpy.arange和numpy.repeat以某种方式完成？

salary=500
for i in range(0,15,1):
    salary=1.1*salary
    print(salary)

Answer 1

使用1.1作为基础创建一个指数数组并使用起始值缩放 -

In [52]: 500*(1.1**np.arange(4)) # 4 is number of output elements
Out[52]: array([500. , 550. , 605. , 665.5])