一维随机漫步1.0

模拟一维随机漫步1.0

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

n=100
steps = np.arange(0,n)
for i in range(n):
    steps[i] = n

for i in range(n):
    for j in range(n):
        steps[i] += choice([1,-1])
#steps = steps.reshape(10,10)
#print(steps)
print(steps)

PI = np.arange(-n,n+1) 
for i in range(2*n+1):
    PI[i] = 0
print(PI)


s=0
for i in range(n):
    PI[steps[i]] = PI[steps[i]] + 1
for i in range(2*n+1):
    s=s+PI[i]

print(PI)

print(s)

a = np.arange(0,2*n+1)  
plt.plot(a,PI,'r')
plt.show()