Python中带边界的矢量化随机游走

import numpy as np

def bounded_random_walk(n_steps, bounds):
    position = np.zeros((n_steps, 2))  # 2维空间中的位置
    position[0] = np.random.uniform(bounds[0], bounds[1], size=2)  # 初始位置

    for i in range(1, n_steps):
        step = np.random.uniform(-1, 1, size=2)  # 随机步长
        new_position = position[i-1] + step

        # 边界检查
        for j in range(2):
            if new_position[j] < bounds[j][0]:
                new_position[j] = bounds[j][0]
            elif new_position[j] > bounds[j][1]:
                new_position[j] = bounds[j][1]

        position[i] = new_position

    return position

# 示例用法
n_steps = 100  # 步数
bounds = [[0, 10], [0, 10]]  # 边界范围

position = bounded_random_walk(n_steps, bounds)
print(position)