模仿醉汉在二维空间上的随机漫步:一个醉汉喝醉酒,每次只能走一步,每步分别沿着x,y轴走一个单位长度,试着画出醉汉的轨迹。
import matplotlib.pyplot as plt
import numpy as np
#起始位置
start=[0,0]
#记录点
move=[start,]
while(len(move)<999):
coordinate=np.array([[0,1],[0,-1],[1,0],[-1,0]])
rd=np.random.randint(len(coordinate))
start=np.array(start)+coordinate[rd]
sign=list(start)
move.append(sign)
move1=np.array(move).T #转置
plt.title('DRUNK')
plt.scatter(0,0,c='r') #起始点标记颜色
plt.plot(move1[0],move1[1],c='y')
plt.scatter(start[0],start[1],c='r')#终点标记颜色
plt.show()
运行结果:
小结:了解一下matplotlib和numpy中的函数