Блог

Предположим: 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]
  

Спасибо!

Вы могли бы достичь этого с помощью базовой нарезки и индексации.

 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)