|
本帖最后由 Hello,World 于 2017-3-30 07:08 编辑
random_walk.py
from random import choice
class RandomWalk():
def __init__(self,num_points=5000):
self.num_points=num_points
self.x_values=[0]
self.y_values=[0]
def fill_work(self):
while len(self.x_values)<self.num_points:
x_dir=choice([1,-1])
X_dis=choice([0,1,2,3,4])
x_step=x_dir*x_dis
y_dir=choice([1,-1])
y_dis=choice([0,1,2,3,4])
y_step=y_dir*y_dis
if x_step==0 and y_step==0:
continue
next_x=self.x_values[-1]+x_step
next_y=self.y_values[-1]+y_step
self.x_values.append(next_x)
self.y_values.append(next_y)
rw_visual.py
import matplotlib.pyplot as plt
from random_walk import RandomWalk
rw=RandomWalk()
rw.fill_work()
plt.scatter(rw.x_values,rw.y_values)
plt.show()
Traceback (most recent call last):
File "rw_visual.py", line 4, in <module>
rw.fill_work()
File "C:\Users\主脑\Desktop\python_work\15生成数据\random_walk.py", line 11, in fill_work
x_step=x_dir*x_dis
NameError: name 'x_dis' is not defined
|
|