python 随机漫步

1. 折线图——通过内置的random模块以纯python的方式实现100步的随机漫步：

import random
position = 0
walk = [position]
steps = 100
for i in range(steps):
    step = 1 if random.randint(0, 1) else -1
    position += step
    walk.append(position)

根据100个随机漫步值生成折线图

import matplotlib.pyplot as plt
plt.plot(walk)
plt.title('Random walk with +1/-1 steps')

 

       2. 一次模拟多个随机漫步

给numpy.random的函数传入一个二元元组产生一个二维数组，一次性计算500个随机漫步过程的累积和

nwalks = 500
nsteps = 100
draws = np.random.randint(0, 2, size=(nwalks, nsteps))
steps = np.where(draws > 0, 1, -1)
walks = steps.cumsum(1)
plt.plot(walks)
plt.title