15_5 重构

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_walk(self):
		while len(self.x_values) < self.num_points:
			x_step = self.get_step()
			y_step = self.get_step()
			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)
			sefl.y_values.append(next_y)
	
	def get_step(self):
		direction = choice([1,-1])
		distance choice([0,1,2,3,4)]
		get_step = direction * distance
		return get_step			

rw_visual.py

import matplotlib.pyplot as plt
from random_walk import RandomWalk

while True:
	rw = RandomWalk(5000)
	rw.fill_walk()
	
	plt.figure(dpi=128, figsize(10,8))
	plt.plot(rw.x_values, rw.y_values,linewidth=5)

	plt.scatter(0, 0, c='green',s=100)
	plt.scatter(rw.x_values[-1],  rw.y_values[-1], c='red', s=100)	

	plt.show()
	
	keep_running = input("Baby one more time!!? (y/n)")
		if keep_running == 'n'
			break