python 项目2 数据可视化2 2021-02-26

import matplotlib.pyplot as plt

from random_walk import RandomWalk

# 只要程序处于活动状态，就不断模拟随机漫步

while True:

    # 创建一个RandomWalk实例

    rw = RandomWalk(50_000)

    rw.fill_walk()

    # 将所有的点都绘制出来

    plt.style.use('classic')

    fig, ax = plt.subplots(figsize=(15, 9))  # 调整尺寸

    point_numbers = range(rw.num_points)

    ax.scatter(rw.x_values, rw.y_values, c=point_numbers,

               cmap=plt.cm.Blues, edgecolor='none', s=1)  # edgecolor删除点周围轮廓

    # 突出起点和终点

    ax.scatter(0, 0, c='green', edgecolors='none', s=100)

    ax.scatter(rw.x_values[-1], rw.y_values[-1], c='red', edgecolors='none',

               s=100)

    # 隐藏坐标轴

    ax.get_xaxis().set_visible(False)

    ax.get_yaxis().set_visible(False)

    plt.show()

    keep_running = input("Make another walk?(y/n):")

    if keep_running == 'n':

        break

from random import choice

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):

        # 计算随机漫步包含的所有点

        # 不断漫步，直到列表到达指定的长度

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

            # 决定前进方向以及沿这个方向前进的距离

            x_direction = choice([1, -1])  # 右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值

            x = self.x_values[-1] + x_step  # 最后一个值相加

            y = self.y_values[-1] + y_step

            self.x_values.append(x)  # 附加到末尾

            self.y_values.append(y)