先创建一个模拟随机漫步的类,再创建一个类的实例。
绿色是起点,红色是终点,颜色越深走的步数越多。
random_walk.py
from random import choice import matplotlib.pyplot as plt class RandomWalk: """一个模拟随机漫步的类""" def __init__(self, walk_numbers=5000): """初始化随机漫步的属性""" self.walk_numbers = walk_numbers #初始位置在(0, 0) self.x_values = [0] self.y_values = [0] def get_step(self): """确定下一步怎么走""" n_direction = choice([1, -1]) n_distance = choice(list(range(10))) n_step = n_direction * n_distance return n_step def fill_walk(self): """随机漫步""" while len(self.x_values) < self.walk_numbers: x_step = self.get_step() y_step = self.get_step() #禁止原地踏步 if x_step * 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) #只要不拒绝就一直模拟 while True: """模拟随机漫步""" rw = RandomWalk(10000) rw.fill_walk() #绘制图形 walk_numbers = list(range(rw.walk_numbers)) plt.figure(figsize=(8, 6)) plt.scatter(rw.x_values, rw.y_values, c = walk_numbers, cmap=plt.cm.Blues, s=10) #标记起点和终点 plt.scatter(0, 0, c='green', s=50) plt.scatter(rw.x_values[-1], rw.y_values[-1], c='red', s=50) #隐藏坐标轴 plt.axes().get_xaxis().set_visible(False) plt.axes().get_yaxis().set_visible(False) plt.show() #是否重复 response = input("Again?(y/n):") if response == 'n': break