根据元素的索引填充NumPy数组

Question

假设： u = 1.2 d = 0.8 n = 3 我想知道如何获得一个包含3（n）个元素的数组，其中每个元素的值为：

上述变量的解决方案是：

[1.2^2 + 0.8^0, 1.2^1 + 0.8^1, 1.2^0 + 0.8^1]

谢谢！

Answer 1

您可以使用Basic slicing and indexing来实现。

import numpy as np

u=1.2
d=0.8
n=3

array=np.zeros(n)

for i in range(n):
    if i + 1 <= n:
        array[i:n]=u**(n-i-1)
        array[i:n] =d**(i)