随机行走轨迹

生成一个1000步的随机行走轨迹, 然后绘制轨迹.

import numpy as np
import matplotlib.pyplot as plt

n_step=1000 #随机行走步数

x_step=np.random.choice([-1,1],n_step)#随机产生1000个1，-1代表X方向每步行走方向
y_step=np.random.choice([-1,1],n_step)#随机产生1000个1，-1代表Y方向每步行走方向

x_trace=np.cumsum(x_step)            #累积求和获得X方向每一步累积的轨迹
y_trace=np.cumsum(y_step)           #累积求和获得Y方向每一步累积的轨迹

plt.plot(x_trace,y_trace)           #画出轨迹
plt.scatter(0,0,c='r',s=100)      #标明起点

plt.scatter(x_trace[-1],y_trace[-1],c='g',s=100)#标明终点