Заполнение массива 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

Этого можно добиться с помощью Базовая нарезка и индексация .

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)