如何通过Python实现蒙特卡罗模拟算法

本文主要介绍蒙特卡罗模拟算法,以及如何通过Python来模拟问题。

文章目录

    • 什么是蒙特卡罗(Monte Carlo)方法?
    • 案例1: π \pi π的计算
      • 如何使用蒙特卡罗方法计算圆周率 π \pi π
      • Python模拟计算
    • 案例2: 定积分的计算
      • 问题描述
      • Python模拟计算
    • 案例3: 排队上厕所问题
      • 问题描述
      • Python模拟
    • 小结

什么是蒙特卡罗(Monte Carlo)方法?

蒙特卡罗(Monte Carlo)方法,又称随机抽样统计试验方法,是通过使用随机数(或更常见的伪随机数)来解决很多计算问题的方法,将所求解的问题同一定的概率模型相联系,用计算机实现统计模拟或抽样,以获得问题的近似解。

蒙特卡罗解题归结为三个主要步骤:

  1. 构造或描述概率过程;
  2. 实现从已知概率分布抽样;
  3. 建立各种估计量。

接下来我们介绍3个简单的案例,看一下如何在实际问题中应用这3个步骤进行求解。

案例1: π \pi π的计算

如何使用蒙特卡罗方法计算圆周率 π \pi π

正方形内部有一个相切的圆,假设圆形的半径为 r r r,那么正方形的边长则为 2 r 2r 2r,因此,圆形和正方形的面积之比为:

π r 2 4 r 2 = π 4 \frac{\pi r^2}{4r^2} = \frac{\pi}{4} 4r2πr2=4π

如何通过Python实现蒙特卡罗模拟算法_第1张图片

现在,在正方形内部,随机产生一系列的点,计算他们与中心点的距离,从而判断是否落在圆的内部,从而:

落 在 圆 内 部 点 数 总 点 数 = π 4 \frac{落在圆内部点数}{总点数} = \frac{\pi}{4} =4π

即可估算出圆周率 π \pi π的大小。

Python模拟计算

Step1 构造或描述概率过程

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.patches import Circle


r = 1  # 半径
a, b = (0.0, 0.0)  # 圆心

# 正方形边界
xmin, xmax = a-1, a+1
ymin, ymax = b-1, b+1

Step2 实现从已知概率分布抽样

由于每个点落入不同位置的概率是一致的,所以我们使用均匀分布进行抽样。

# 随机抽取n个点
n = 10000

# 在正方形范围内,随机抽取n个点
x = np.random.uniform(xmin, xmax, n)
y = np.random.uniform(ymin, ymax, n)

Step3 建立各种估计量

# 计算每个点到圆心的距离
d = np.sqrt((x - a) ** 2 + (y - b) ** 2)
# 圆内点的个数
res = sum(np.where(d<r, 1, 0))  # 距离小于半径即为圆内

# pi
pi = res / n * 4

模拟结果:

3.1572

可以看到,模拟结果和 π \pi π还是比较接近的,我们可以通过增大样本点的个数 n n n,使得这个估计结果更加精确。

把所有点画出来:

# 绘制正方形
fig = plt.figure(figsize=(10, 10))
axes = fig.add_subplot(1, 1, 1)
plt.plot(x, y, 'ro', markersize=1)
plt.axis('equal')

# 绘制圆形
circle = Circle(xy=(a, b), radius=r, alpha=0.5, color='gray')
axes.add_patch(circle)
plt.grid(True, linestyle='--', linewidth=0.8)

如何通过Python实现蒙特卡罗模拟算法_第2张图片

案例2: 定积分的计算

问题描述

计算 y = x 2 y=x^2 y=x2在[0, 1]区间的定积分,也就是下图中的阴影面积大小。

按照蒙特卡罗模拟的思想,我们可以计算有多少点落在积分范围内(判断条件高度 y < x 2 y \lt x^2 y<x2),落在阴影范围内的点数跟所有抽样点数的比值就是所要求的积分值。

如何通过Python实现蒙特卡罗模拟算法_第3张图片

Python模拟计算

Step1 构造或描述概率过程

# 正方形边界
xmin, xmax = 0.0, 1.0
ymin, ymax = 0.0, 1.0

# 函数曲线
def f(x):
    return x ** 2

Step2 实现从已知概率分布抽样

由于每个点落入不同位置的概率是一致的,所以我们使用均匀分布进行抽样。

# 随机抽取n个点
n = 10000

# 在正方形范围内,随机抽取n个点(均匀分布)
x = np.random.uniform(xmin, xmax, n)
y = np.random.uniform(ymin, ymax, n)

Step3 建立各种估计量

# 统计落在函数图像下方点的数目
res = sum(np.where(y < f(x), 1, 0)) 

# 计算积分的近似值
integral = res / n

模拟结果:

0.3302

我们实际计算一下,跟模拟结果进行对比:

∫ 0 1 x 2 d x = 1 3 x 3 ∣ 0 1 = 1 3 \int_0^1 x^2dx = \frac{1}{3}x^3\bigg|_0^1 = \frac{1}{3} 01x2dx=31x301=31

可以看到模拟结果还是相对准确的。

把所有点画出来:

# 绘制正方形
fig = plt.figure(figsize=(10, 10))
axes = fig.add_subplot(1, 1, 1)
plt.plot(x, y, 'ro', markersize=1)
plt.axis('equal')

# 绘制函数曲线
xi = np.linspace(0, 1, 100)
yi = xi ** 2
plt.plot(xi, yi, '--k')
plt.fill_between(xi, yi, 0, color='gray', alpha=0.5, label='area')  # 填充曲线下的区域

如何通过Python实现蒙特卡罗模拟算法_第4张图片

案例3: 排队上厕所问题

问题描述

一场电影结束后会有20个人想要上厕所,假设只有一个厕所,同时:

  • 这20个人会在0-10分钟内全部到达厕所;
  • 每个人上厕所的时间在1-3分钟之内;

求开始时间、结束时间、等待时间、上厕所时长和厕所空闲时间分布情况。

Python模拟

Step1 构造或描述概率过程

starting_time = [0] * num  # 每个人的开始时间
finish_time = [0] * num  # 每个人的结束时间
waiting_time = [0] * num  # 每个人的等待时间
empty_time = [0] * num  # 厕所的空闲时间

Step2 实现从已知概率分布抽样

num = 20  # 人数
arriving_time = np.random.uniform(0, 10, size=num)  # 到达厕所时长分布
arriving_time.sort()  # 按照到达厕所时间从小到大排序
working_time = np.random.uniform(1, 3, size=num)  # 上厕所时长分布

Step3 建立各种估计量

# 初始化第一个数据
starting_time[0] = arriving_time[0]
finish_time[0] = starting_time[0] + working_time[0]
waiting_time[0] = starting_time[0] - arriving_time[0]  # 0

for i in range(1, num):
    last_people_finish_time = finish_time[i-1]
    # 到达时间早于上一个人的结束时间
    if arriving_time[i] < last_people_finish_time:
        starting_time[i] = last_people_finish_time
        waiting_time[i] = last_people_finish_time - arriving_time[i]
        empty_time[i] = 0
    # 到达时间晚于等于上一个人的结束时间
    else:
        starting_time[i] = arriving_time[i]
        waiting_time[i] = 0
        empty_time[i] = arriving_time[i] - last_people_finish_time
    finish_time[i] = starting_time[i] + working_time[i]

模拟结果可视化:

# 等待曲线
plt.plot(starting_time, label='starting')
plt.plot(finish_time, label='finish')
plt.plot(waiting_time, label='waiting')
plt.plot(empty_time, label='empty')
plt.plot(working_time, label='working')

plt.legend()

如何通过Python实现蒙特卡罗模拟算法_第5张图片

小结

本文首先介绍蒙特卡罗模拟算法的思想,及其求解问题的3个步骤。

接着,通过3个简单的案例讲解了如何使用Python实现蒙特卡罗模拟算法。

说明:本文问题来源于网易云课堂的数据分析师(python)课程。

你可能感兴趣的:(概率统计,python,算法)