如何用Python循环模拟5个一维随机游走并在同图绘制？

问题说明 #

需要模拟5个一维随机游走并将所有路径绘制在同一张图中，单条游走的代码运行正常，但尝试的多路径模拟代码无法实现预期效果。

原尝试代码的问题 #

1. 提前返回导致循环中断：内层for i in range(n)循环里的return x, y会在第一次迭代时就直接返回，仅生成了一条路径的第一步就结束函数，根本无法完成5个游走的模拟。
2. 路径变量未独立初始化：所有游走共用同一个x、y列表，没有为每个游走单独创建新的路径起点，会导致路径混乱叠加。

修改后的完整代码 #

方式一：生成一条路径就绘制一条

import numpy as np
import matplotlib.pyplot as plt

a = -1
b = 1

def random_walk(n):
    x = [0]
    y = [0]
    for i in range(n):
        r = a + (b - a) * np.random.rand()
        step = y[i] + (5/3) * r
        xstep = x[i] + 1
        y.append(step)
        x.append(xstep)
    return x, y

# 模拟5个随机游走并绘制
plt.title("5条一维随机游走路径")
plt.xlabel("步数")
plt.ylabel("位置")

for j in range(5):
    x, y = random_walk(200)
    plt.plot(x, y, '.-', label=f"游走{j+1}")

plt.legend()
plt.show()

方式二：先存储所有路径再统一绘制

import numpy as np
import matplotlib.pyplot as plt

a = -1
b = 1

def random_walk(n):
    x = [0]
    y = [0]
    for i in range(n):
        r = a + (b - a) * np.random.rand()
        step = y[i] + (5/3) * r
        xstep = x[i] + 1
        y.append(step)
        x.append(xstep)
    return x, y

# 存储所有路径
walks = []
for _ in range(5):
    walks.append(random_walk(200))

# 绘制所有路径
plt.title("5条一维随机游走路径")
plt.xlabel("步数")
plt.ylabel("位置")

for idx, (x, y) in enumerate(walks):
    plt.plot(x, y, '.-', label=f"游走{idx+1}")

plt.legend()
plt.show()

关键修改点说明 #

• 保留了原本正常的单条游走生成函数random_walk，通过外层循环调用它5次，每次生成独立的路径。
• 移除了原代码中错误的提前return语句，确保每条路径能完整生成n步。
• 为每条游走添加了图例标签，方便区分不同路径。
• 统一设置了图表的标题、坐标轴标签，让可视化更清晰。

内容的提问来源于stack exchange，提问作者Hitchens