[Python机器学习]Nagel-Schreckenberg(交通流)模型

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @File  : Nagel-Schreckenberg(交通流)模型.py
# @Author: 赵路仓
# @Date  : 2020/3/1
# @Desc  :
# @Contact : 398333404@qq.com

import matplotlib
from matplotlib.font_manager import *
import matplotlib.pyplot as plt
import numpy as np

plt.rcParams[‘font.sans-serif‘] = [‘SimHei‘]
matplotlib.rcParams[‘axes.unicode_minus‘] = False
np.random.seed(0)


def Run(path=5000, n=100, v0=60, ltv=120, p=0.3, times=3000):
    path = 5000.0  # 道路长度
    n = 100  # 车辆数目
    v0 = 60  # 初始速度
    ltv = 120  # 最大限速
    p = 0.3  # 减速概率
    times = 3000  # 模拟的时刻数目

    # x保存每辆车在道路上的位置，随机初始化
    x = np.random.rand(n) * path
    x.sort()
    # v保存每辆车的速度，初速度相同
    v = np.ones(n) * v0

    plt.figure(figsize=(5, 4), facecolor=‘w‘)
    # 模拟每个时刻
    for t in range(times):
        plt.scatter(x, [t] * n, s=1, c=‘k‘, alpha=0.05)
        # 模拟每辆车
        for i in range(n):
            # 计算当前车与前车的距离，注意是环形车道
            if x[(i + 1) % n] > x[i]:
                d = x[(i + 1) % n] - x[i]
            else:
                d = path - x[i] + x[(i + 1) % n]
            # 根据距离计算下一秒的速度
            if v[i] < d:
                if np.random.rand() > p:
                    v[i] += 1
                else:
                    v[i] -= 1
            else:
                v[i] = d - 1
        # 对速度进行限制
        v = v.clip(0, ltv)

        # 一秒后，车辆的位置发生了变化
        x += v
        # 注意是环形车道
        x = x % path

    # 展示
    plt.xlim(0, path)
    plt.ylim(0, times)
    plt.xlabel(u‘车辆位置‘)
    plt.ylabel(u‘模拟时间‘)
    plt.title(u‘交通模拟(车道长度%d,车辆数%d,初速度%s,减速概率%s)‘ % (path, n, v0, p))
    # plt.tight_layout(pad=2)
    plt.show()


if __name__ == ‘__main__‘:
    # Run(p=0.3)
    Run(p=0.2)