from random import choice
class Randomwalk ():
  '''一個(gè)生成隨機(jī)數(shù)漫步的類'''
  def __init__(self,num_point=5000):
    '''初始化隨機(jī)漫步的屬性'''
    self.num_point = num_point
    #所有隨機(jī)漫步的開(kāi)始都是坐標(biāo)[0,0]
    self.x_lab = [0]
    self.y_lab = [0]
  def fill_walk(self):
    '''計(jì)算隨機(jī)漫步的所有點(diǎn)'''
    while len(self.x_lab) < self.num_point:
      #決定前進(jìn)方向以及前進(jìn)的距離
      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
      #拒絕原地不動(dòng)
      if x_step == 0 and y_step == 0:
        continue
      #計(jì)算下一個(gè)點(diǎn)X和Y的值
      next_x = self.x_lab[-1] + x_step
      next_y = self.y_lab[-1] + y_step
      self.x_lab.append(next_x)
      self.y_lab.append(next_y)

import matplotlib.pyplot as plt
from random_walk import Randomwalk
from random import choice
rw = Randomwalk()
rw.fill_walk()
plt.scatter(rw.x_lab,rw.y_lab,s=15)
plt.show()

import matplotlib.pyplot as plt
from random_walk import Randomwalk
from random import choice
while True:
  rw = Randomwalk()
  rw.fill_walk()
  #設(shè)置繪畫(huà)窗口大小
  plt.figure(dpi=128,figsize=(10,6))
  point_numbers = list(range(rw.num_point))
  #突出起點(diǎn)（0,0）和終點(diǎn)
  plt.scatter(0,0,c='green',edgecolors='none',s=100)
  plt.scatter(rw.x_lab[-1],rw.y_lab[-1],c='red',edgecolors='none',s=100)
  #隱藏坐標(biāo)軸
  plt.axes().get_xaxis().set_visible(False)
  plt.axes().get_yaxis().set_visible(False)
  plt.scatter(rw.x_lab,rw.y_lab,c=point_numbers,cmap=plt.cm.Blues,edgecolors='none',s=15)
  plt.show()
  keep_running = input("Make another walk?(y/n): ")
  keep_running = keep_running.lower()
  if keep_running == 'n':
    break