使用Python/Pandas/Numpy的几何级数（无循环，使用递归）

Question

我想用Python/Pandas/Numpy实现一个几何级数。在

以下是我所做的：

N = 10
n0 = 0
n_array = np.arange(n0, n0 + N, 1)
u = pd.Series(index = n_array)
un0 = 1
u[n0] = un0
for n in u.index[1::]:
    #u[n] = u[n-1] + 1.2 # arithmetic progression
    u[n] = u[n-1] * 1.2 # geometric progression
print(u)

我得到：

^{pr2}$

我想知道如何避免使用这个for循环。在

我看了看 https://fr.wikipedia.org/wiki/Suite_g%C3%A9om%C3%A9trique 并发现uün可以表示为：uüun=u{n_0}*q^{n-n}0}

所以我就这么做了

n0 = 0
N = 10
n_array = np.arange(n0, n0 + N, 1)
un0 = 1
q = 1.2
u = pd.Series(map(lambda n: un0 * q ** (n - n0), n_array), index = n_array)

没关系。。。但我正在寻找一种方法来定义它

u_n0 = 1
u_n = u_{n-1} * 1.2

但是我不知道如何使用Python/Pandas/Numpy来实现。。。我想知道是否可能。在

Answer 1

以下是我在《熊猫》系列中的表现：

N = 10
n0 = 0
n_array = np.arange(n0, n0 + N, 1)
u = pd.Series(index = n_array)
u[n0] = 1
q = 1.2
# option 1:
u = pd.Series(u[n0]*q**(u.index.values - n0), index = n_array)
# or option 2 with cumprod
u[1:] = q
u = u.cumprod()