matplotlib之随机漫步

 # 随机漫步类
 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()

运行截图：