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

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 开始？

How do I initialise the plot of my function to start at 0?

python

numpy

matplotlib

random-walk