# 随机漫步类

 from random import choice

 from matplotlib import pyplot as plt

 from pylab import mpl

 from matplotlib import rcParams

 class RandomWalk():

     '''生成随机漫步的类'''

     def __init__(self, total_num=15000):

         self.total_num = total_num

         '''初始坐标（0,0）'''

         self.x_values = [0]

         self.y_values = [0]

     def get_step(self):

         '''随机生成移动的方向和距离'''

         diretion = choice([1, -1])

         distance = choice([0, 1, 2, 3, 4, 5])

         return diretion * distance

     def fill_walk(self):

         '''计算随机漫步中所有的点'''

         while len(self.x_values) < self.total_num:

             '''计算前进方向和距离'''

             x_step = self.get_step()

             y_step = self.get_step()

             '''拒绝原地踏步'''

             if x_step == 0 and y_step == 0:

                 continue

             '''计算下一个点的x和y的值'''

             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)

 # 设置默认字体，解决中文显示乱码问题

 mpl.rcParams['font.sans-serif'] = ['SimHei']

 # 解决负号'-'显示为方块的问题

 rcParams['axes.unicode_minus'] = False

 # 随机漫步实例

 rw = RandomWalk()

 rw.fill_walk()

 point_nunbers = list(range(rw.total_num))

 # 调整绘图窗口尺寸，单位为英寸

 plt.figure(figsize=(8, 6))

 # 给点着色显示

 plt.scatter(rw.x_values, rw.y_values, c=point_nunbers, cmap=plt.cm.Blues, s=1)

 plt.title("随机漫步图", fontsize=18)

 # 突出起点和终点

 plt.scatter(0, 0, c='green', s=30)

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

 # 隐藏坐标轴

 # plt.axes().get_xaxis().set_visible(False)

 # plt.axes().get_yaxis().set_visible(False)

 plt.show()