python中一维随机游动的均方位移

import random

def one_dimensional_random_walk(steps):
    position = 0  # 初始位置为0
    displacement = 0  # 位移初始化为0

    for _ in range(steps):
        step = random.choice([-1, 1])  # 随机选择向左或向右移动
        position += step
        displacement += step ** 2  # 累计位移的平方

    return displacement

steps = 1000  # 游动步数
mean_square_displacement = one_dimensional_random_walk(steps) / steps

print("一维随机游动的均方位移：", mean_square_displacement)