Question

这可能是一个非常愚蠢的问题，但是由于某种原因，我似乎无法解决该问题。我在这里创建了一个用于随机游走的函数，该函数仅在一个试验中使用numpy二项式函数（即，如果它小于0.5，则为-1，大于+1。但是，这显然会使函数的第一个值成为-1或+1，而我希望它为零。如何更改？

import numpy as np
import matplotlib.pyplot as plt

def random_walk(N,d):
    walk = d*np.cumsum(2*np.random.binomial(1,.5,N-1)-1)
    return walk

plt.plot(np.arange(1,250),random_walk(250,1))
plt.show()

再次，这可能真的很简单，我真的很傻，但是我非常感谢您的帮助！

Answer 1

微调您的代码。希望这就是您要寻找的。

import numpy as np
import matplotlib.pyplot as plt

def random_walk(N,d):
    walk = np.concatenate(([0],np.cumsum(2*np.random.binomial(1,.5,N-1)-1)))
    return walk

plt.plot(np.arange(0,250),random_walk(250,1))
plt.show()

如何初始化函数从0开始的图？

1 个答案: