Python-随机漫步序列(random,Matplotlib)

每次移动的方向和距离都通过random中的choice()决定，通过Matplotlib显示轨迹。

可以使用字母y循环画图

代码：

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])
            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=self.x_values[-1]+x_step
            y=self.y_values[-1]+y_step
            #保存位置到列表中
            self.x_values.append(x)
            self.y_values.append(y)


'''绘制随机漫步图'''
import matplotlib.pyplot as plt

while True:
    #创建一个RandomWalk实例
    rW=RandomWalk(50_000)
    rW.fill_walk()
    #绘制所有点
    plt.style.use('classic')
    fig,ax=plt.subplots(figsize=(15,9))
    # 通过range生成一个数字序列
    point_numbers=range(rW.num_points) #绘制漫步轨迹
    ax.scatter(rW.x_values,rW.y_values,s=15,c=point_numbers,cmap=plt.cm.Blues,edgecolors='none')
    # 绘制起点和终点
    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("Run again?(y/n):")
    if keep_running == 'n':
        break

结论：

参考资料《python编程：从入门到实践》