运用matplotlib写一个随机漫步

"""随机漫步"""

from random import choice
import matplotlib.pyplot as plt

class RandomWalk():

    def __init__(self, num_points=5000):
        # 初始化随机漫步的属性
        self.num_points = num_points
        # 设置随机漫步都始于(0,0)
        self.x_values = [0]
        self.y_values = [0]

    def fill_walk(self):

        # 不停漫步。直到达到5000
        while len(self.x_values) < self.num_points:
            #决定前进的方向
            x_direction = choice([1, -1])
            #决定行走的距离
            x_distance = choice([0,1,2,3,4])
            x_step = x_direction*x_distance

            #决定前进的方向
            y_direction = choice([1, -1])
            #决定行走的距离
            y_distance = choice([0,1,2,3,4])
            y_step = y_direction*y_distance


            #拒绝原地踏步
            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)


#实例化对象
##rw = RandomWalk()
##rw.fill_walk()
##plt.scatter(rw.x_values, rw.y_values, s=15)
##plt.savefig("./t2.png")  保存图片
##plt.show()


while True:
    rw = RandomWalk()
    rw.fill_walk()
    plt.scatter(rw.x_values, rw.y_values, s=15)
    plt.show()

    keep_running = input("是否进行下一个漫步：(y/n)")
    if keep_running == "n":
        break