python.matplotlib/random.choice学习范例-随机生成点图

from random import choice

class RandomWark():

    '''生成一个随机漫步数据的类'''

    

    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)

        #x，y的前进方向以及沿这个方向的距离

            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)

import matplotlib.pyplot as plt

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

while True:

    #创建一个实例，并绘制点

    rw=RandomWark()

    rw.fill_walk()

#调整尺寸适应屏幕

    plt.figure(dpi=128,figsize=(10,6))

#设置每个漫步点的颜色

    point_numbers=list(range(rw.num_points))

    plt.scatter(rw.x_values,rw.y_values,c=point_numbers,cmap=plt.cm.Blues,

        edgecolor='none',s=15)

    

#标记起点和终点

    plt.scatter(0,0,c='green',edgecolor='none',s=100)

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

#隐藏坐标轴

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

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

    plt.show()

    

#判断进程

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

    if keep_running == 'n':

        break