模拟一维随机漫步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()