基于Python条件语句的相关随机游走实现问题咨询

你尝试用Python实现1D相关随机游走，期望游走者以相关系数a保持同方向移动，设置了50步、1000条轨迹，但代码逻辑存在问题，导致无法得到预期的双峰值归一化直方图。以下是问题分析和修正方案：

原代码的核心问题 #

1. 相关系数a的定义位置冗余：在循环内部重复赋值a=0.7，既没必要也不利于参数调整
2. 方向判断逻辑错误：原条件分支没有准确实现“以概率a保持当前方向，概率1-a反转方向”的核心规则，导致游走行为不符合预期

修正后的代码 #

import numpy as np
import matplotlib.pyplot as plt

def RW_1D(n, N, a=0.7):
    # n: 总步数, N: 轨迹数量, a: 同方向保持概率
    X = np.zeros((n, N))
    
    # 初始化每条轨迹的第一步移动方向
    first_steps = np.random.choice([1, -1], size=N)
    X[0, :] = first_steps
    
    for j in range(N):
        xnow = X[0, j]
        for i in range(1, n):
            # 获取上一步的移动方向
            prev_direction = X[i-1, j] - (X[i-2, j] if i > 1 else 0)
            # 以概率a保持原方向，1-a反转方向
            new_direction = prev_direction if np.random.rand() < a else -prev_direction
            xnew = xnow + new_direction
            X[i, j] = xnew
            xnow = xnew
    
    return X

# 生成轨迹数据
X = RW_1D(50, 1000, a=0.7)
# 提取所有轨迹的最终位置
final_positions = X[-1, :]

# 绘制归一化直方图
plt.hist(final_positions, bins='auto', density=True, alpha=0.7, color='#1f77b4')
plt.xlabel('最终位置')
plt.ylabel('归一化概率密度')
plt.title('相关随机游走最终位置分布 (a=0.7)')
plt.grid(axis='y', alpha=0.75)
plt.show()

修正说明 #

• 将a作为函数参数传入，方便灵活调整相关系数，避免冗余赋值
• 第一步方向生成用np.random.choice简化实现，高效直观
• 核心逻辑改为：记录上一步的移动方向，以概率a保持该方向，否则反转方向，完全匹配相关随机游走的定义
• 补充了直方图绘制代码，直接展示最终位置的分布，可观察到预期的双峰值效果

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