首页 > 编程 > Python > 正文

Python模拟随机游走图形效果示例

2020-01-04 15:57:00
字体:
来源:转载
供稿:网友

本文实例讲述了Python模拟随机游走图形效果。分享给大家供大家参考,具体如下:

在python中,可以利用数组操作来模拟随机游走。

下面是一个单一的200步随机游走的例子,从0开始,步长为1和-1,且以相等的概率出现。纯Python方式实现,使用了内建的 random 模块:

# 随机游走import matplotlib.pyplot as pltimport randomposition = 0walk = [position]steps = 200for i in range(steps):  step = 1 if random.randint(0, 1) else -1  position += step  walk.append(position)fig = plt.figure()plt.title("www.vevb.com")ax = fig.add_subplot(111)ax.plot(walk)plt.show()

Python,模拟,随机游走,图形

第二种方式:简单的把随机步长累积起来并且可以可以使用一个数组表达式来计算。因此,我用 np.random 模块去200次硬币翻转,设置它们为1和-1,并计算累计和:

# 随机游走import matplotlib.pyplot as pltimport numpy as npnsteps = 200draws = np.random.randint(0, 2, size=nsteps)steps = np.where(draws > 0, 1, -1)walk = steps.cumsum()fig = plt.figure()plt.title("www.vevb.com")ax = fig.add_subplot(111)ax.plot(walk)plt.show()

Python,模拟,随机游走,图形

一次模拟多个随机游走

# 随机游走import matplotlib.pyplot as pltimport numpy as npnwalks = 5nsteps = 200draws = np.random.randint(0, 2, size=(nwalks, nsteps)) # 0 or 1steps = np.where(draws > 0, 1, -1)walks = steps.cumsum(1)fig = plt.figure()plt.title("www.vevb.com")ax = fig.add_subplot(111)for i in range(nwalks):  ax.plot(walks[i])plt.show()

Python,模拟,随机游走,图形

当然,还可以大胆的试验其它的分布的步长,而不是相等大小的硬币翻转。你只需要使用一个不同的随机数生成函数,如 normal 来产生相同均值和标准偏差的正态分布:

steps = np.random.normal(loc=0, scale=0.25, size=(nwalks, nsteps))

Python,模拟,随机游走,图形

 

希望本文所述对大家Python程序设计有所帮助。


注:相关教程知识阅读请移步到python教程频道。
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表