简单随机漫步

最近在看用Python进行数据分析这本书，里面提到了随机漫步，所以试了试。

今天把代码放这里，以后随时可以看看。

##################一条随机漫步线###############################

import random
import numpy as np
position=0
walk=[position]
steps=1000
for i in np.arange(steps):
    step=1 if random.randint(0,1) else -1
    position+=step
    walk.append(position)
import matplotlib.pyplot as plt
plt.scatter(list(np.arange(steps)), walk[0:1000],cmap=plt.cm.Reds, edgecolors='none', s=5)
plt.scatter(0, 0, c='green', edgecolors='none', s=100)
plt.scatter(list(np.arange(steps))[-1],walk[-1], c='red', edgecolors='none', s=100)
plt.title(u"我们家王震小可爱学走路",fontproperties='SimHei',fontsize=20)
plt.xlabel(u"步数",fontproperties='SimHei',fontsize=20)
plt.ylabel(u"距离",fontproperties='SimHei',fontsize=20)
plt.show()

########################同时进行5000个随机漫步，选择5条进行绘图###############################

import numpy as np
from numpy import random
nwalks=5000
nsteps=1000
draws=np.random.randint(0,2,size=(nwalks,nsteps))
steps=np.where(draws>0,1,-1)
walks=steps.cumsum(1)
walks=walks[0:5]
import pylab as pl
linestyle=['cx--','mo:','kp-','bs--','kp-']#红，绿，黄，蓝，粉,每个折线给不同的颜色
x=[i for i in np.arange(0,len(walks[0]))]#横坐标
linenum=len(walks)
plotlist=[] 
for i in np.arange(0,linenum):
    plot1,=pl.plot(x,walks[i],linestyle[i])#画出对应折线图
    plotlist.append(plot1)
plt.title(u"小可爱学走路",fontproperties='SimHei',fontsize=20)
plt.xlabel(u"步数",fontproperties='SimHei',fontsize=20)
plt.ylabel(u"距离",fontproperties='SimHei',fontsize=20)
pl.show()

猜你喜欢