---10000步的可视化效果
# %% cell 开始 cell可以理解为文章中的段落
# "#%%"表示一级(最高级), "#%%%"表示二级, "#%%%%"表示三级, 以此类推.
# random walk
wlk=np.random.randint(0,2,size=(2,3000))
wlk=np.where(wlk>0,1,-1)
wlk
wlk_cumsum=wlk.cumsum(axis=1)
plt.scatter(wlk_cumsum[0],wlk_cumsum[1],color='purple')
dist=np.sqrt((wlk_cumsum[0]-wlk[0][0])**2+(wlk_cumsum[1]-wlk[1][0])**2)
dist
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.plot(wlk[0][0],wlk[1][0],color='yellow',marker='*',markersize=19)
plt.plot(wlk_cumsum[0][-1],wlk_cumsum[1][-1],color='red',marker='*',markersize=19)
plt.plot(wlk_cumsum[0][dist.argmax()],wlk_cumsum[1][dist.argmax()],color='blue',marker='*',markersize=19)
plt.title('yellow=start, red=end，blue=max dist')
# %% cell 结束