今天给大家分享用Python matplotlib来写随机漫步的小程序
先安装两个库:
pip install matplotlib
pip install numpy
引用库:
import matplotlib.pyplot as mp
import numpy as np
一:只向四个方向随机漫步
def random_walking(x):#定义方法
a=[[1,0],[0,1],[-1,0],[0,-1]]#定义四个方向
l=np.random.choice([0,1,2,3])#随机选取四个数字
x+=np.array(a[l])#随机选取一个方向
return x
x,y=[0],[0]
for i in range(1000):
l=random_walking(np.array([x[-1],y[-1]]))
x+=[1[0]]
y+=[1[1]]
mp.plot(x,y,color="g")
mp.show()
每次运行都是随机结果:
二:随机任意方向漫步
方法都和上面的差不多,只不过是随机任意角度
直接看代码:
def Any_Angle_Walking(w):
t=2*np.pi*np.random.random()
w+=np.array([np.cos(t),np.sin(t)])
return w
x,y=[0.0],[0.0]#从坐标原点开始
for i in range(1000):#随机漫步1000次
temp=Any_Angle_Walking(np.array([x[-1],y[-1]]))
x+=[temp[0]]
y+=[temp[1]]
mp.plot(x,y,color="g")
mp.show()
好了,就分享这莫点了,谢谢大家的支持!