简单随机漫步

小灰灰 2022-06-03 10:53 144阅读 0赞

最近在看用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()

![Center][]

\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#同时进行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()

![Center 1][]

[Center]: /images/20220603/c4d033b152894855b1560ef1f102fd38.png
[Center 1]: /images/20220603/7be9e3cf21e643569be1ce0a73ae8ecd.png