从 R 转换为 python 时列表索引出错

问题描述

我正在尝试转换我的 R 代码

set.seed(2021)
startn <- 10

j <- 1
maxj <- 10^5
n <- startn
time <- 0


while (n[j] > 0 & j < maxj){
   time[j+1] <- time[j] + rexp(1,rate=2*n[j]^2/5)  
   n[j+1] <- n[j] + sample(c(-1,1),1)
   j <- j+1
   }

plot(time,n,xlim=c(0,max(time)),ylim=c(0,max(n)))

进入蟒蛇

n_0 = 10

j = 0
maxj = 10^5
n = [n_0]
time = [0]


while n[j] > 0 & j < maxj:
    time[j+1] = time[j] + np.random.exponential(2*n[j]**2/5) 
    n[j+1] = n[j] + np.random.choice([-1,1])
    j = j+1

但由于错误列表分配索引超出范围而失败。有人能指出我的错误吗？

解决方法

我不确定这是否是您需要的，但这解决了索引问题。请尝试：

import numpy as np

n_0 = 10
j = 0
maxj = 10^5
n = [n_0]
time = [0]

while n[j] > 0 & j < maxj:
    print(time,n)
    time.append(np.random.exponential(2*n[j]**2/5)) 
    n.append(np.random.choice([-1,1]))
    j = j+1