python - 100 步 100 名随机游走者的平均值

Question

我被要求编写 100 个随机行走程序的代码，每个程序有 1000 步。然后绘制 100 名步行者的平均步数。我能够在一个图中绘制所有步行者，但找不到绘制平均值的方法。任何帮助将不胜感激。谢谢。

这是我的代码:

import numpy as np
import matplotlib.pyplot as plt
import random

# 1
N = 100
for j in range(N):
    def randomwalk1D(n):
        x, t = 0, 0
        # Generate the time points [1, 2, 3, ... , n]
        time = np.arange(n + 1)
        position = [x]
        directions = [1, -1]
        for i in range(n):
            # Randomly select either +1 or -1
            step = np.random.choice(directions)
            
            # Move the object up or down
            if step == 1:
                x += 1
            elif step == -1:
                x -= 1
            # Keep track of the positions
            position.append(x)
        return time, position
    rw = randomwalk1D(1000)
    plt.plot(rw[0], rw[1], 'r-', label="rw")
plt.show()

Answer 1

像这样修改你的代码:

import numpy as np
import matplotlib.pyplot as plt
import random

# 1
N = 100
walkers = []  # HERE
for j in range(N):
    def randomwalk1D(n):
        x, t = 0, 0
        # Generate the time points [1, 2, 3, ... , n]
        time = np.arange(n + 1)
        position = [x]
        directions = [1, -1]
        for i in range(n):
            # Randomly select either +1 or -1
            step = np.random.choice(directions)
            
            # Move the object up or down
            if step == 1:
                x += 1
            elif step == -1:
                x -= 1
            # Keep track of the positions
            position.append(x)
        return time, position
    rw = randomwalk1D(1000)
    walkers.append(rw)
    plt.plot(rw[0], rw[1], 'r-', label="rw")
walkers = np.array(walkers)  # HERE
plt.plot(walkers[0][0], walkers[:, 1].mean(axis=0), 'b-', label="mean")  # HERE
plt.show()

更新

How can I calculate the root mean square of this array?

# Remove the time dimension, extract the position
arr = walkers[:, 1]

# Compute the Root Mean Square
rms = np.sqrt(np.mean(arr**2, axis=1))