用random.choice完成随机漫步绘图！
今天学习了用random.choice完成随机漫步绘图。
首先，调用random.choice函数。
from random import choice
import matplotlib.pyplot as plt
创建一个随机漫步的类
class RandomWalk():
#初始化随机漫步的属性
def __init__(self, num_points = 5000):
self.num_points = num_points
self.x_values = x_values
self.y_values = y_values
def fill_walk(self):
while len(self.x_values) < 5000:
#定义一个函数，设置随机漫步的方向距离大小
#设置x轴的方向距离大小
x_directions = [1, -1]
x_distance = [0, 1, 2, 3, 4, 5]
x_step = x_directions * x_distance
#设置y轴方向距离大小
y_directions = [1, -1]
y_distance = [0, 1, 2, 3, 4, 5]
y_step = y_directions * y_distance
#禁止原地踏步
if x_step==0 and y_step==0:
continue
#计算下一步的位置
next_x = self.x_values[-1] + x_step
next_y = self.y_values[-1] + y_step
self.x_valuer.append(next_x)
self.y_values.append(next_y)
rw = RandomWalk()
rw.fill_walk()
plt.scatter(rw.x_values, rw.y_values, c=rw.y_values, cmap=cm.plt.cm.Reds, s=5)
plt.show()
代码中使用了颜色映射，便于查看。
还可以往其中加入自动保存和输出rw.x_values和rw.y_values列表数据。
以上就是全部代码，有错误请指正，大佬勿喷。
以下是生成的数据图。
模拟多次随机，并给点着色
from random import choice
import matplotlib.pyplot as plt
创建一个随机漫步的类
class RandomWalk():
#初始化随机漫步的属性
def __init__(self, num_points = 5000):
self.num_points = num_points
self.x_values = x_values
self.y_values = y_values
def fill_walk(self):
while len(self.x_values) < 5000:
#定义一个函数，设置随机漫步的方向距离大小
#设置x轴的方向距离大小
x_directions = [1, -1]
x_distance = [0, 1, 2, 3, 4, 5]
x_step = x_directions * x_distance
#设置y轴方向距离大小
y_directions = [1, -1]
y_distance = [0, 1, 2, 3, 4, 5]
y_step = y_directions * y_distance
#禁止原地踏步
if x_step==0 and y_step==0:
continue
#计算下一步的位置
next_x = self.x_values[-1] + x_step
next_y = self.y_values[-1] + y_step
self.x_valuer.append(next_x)
self.y_values.append(next_y)
rw = RandomWalk()
rw.fill_walk()
#给点着色
number_points = list(range(rw.num_points))
plt.scatter(rw.x_values, rw.y_values, c=number_points, cmap=cm.plt.cm.Reds, s=5)
plt.show()
#设置多次随机漫步
keep_running = input("Make another walk? (Y/N)")
if keep_running=='N':
break
隐藏坐标轴
#隐藏坐标轴
plt.axes().get_xaxis().set_visible(False)
plt.axes().get_yaxis().set_visible(False)
增加点数
rw.RandomWalk(num_points=50000)
调整尺寸以适合屏幕
plt.figure(dpi=72, figsize=(10, 10))
改进的随机漫步
from random import choice
import matplotlib.pyplot as plt
'''生成一个随机漫步的类'''
class RandomWalk():
def __init__(self, num_points=5000):
self.num_points = num_points
self.x_values = [0]
self.y_values = [0]
def fill_walk(self):
while len(self.x_values) < self.num_points:
x_step = self.get_step()
y_step = self.get_step()
if x_step==0 and y_step==0:
continue
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)
def get_step(self):
#设置方向距离大小
directions = choice([1, -1])
distance = choice([0, 1, 2, 3, 4])
return directions * distance
while True:
rw = RandomWalk(num_points=5000)
rw.fill_walk()
plt.figure(num='figure0', dpi=72, figsize=(6, 6))
numbers_points = list(range(rw.num_points))
#plt.plot(rw.x_values, rw.y_values, linewidth=0.1)
plt.scatter(rw.x_values, rw.y_values, s=5, edgecolors='none',
c=numbers_points, cmap=plt.cm.Blues)
plt.scatter(0, 0, c='yellow', edgecolor='none', s=20)
plt.scatter(rw.x_values[-1], rw.y_values[-1], c='red', edgecolor='none', s=20)
plt.axes().get_xaxis().set_visible(False)
plt.axes().get_yaxis().set_visible(False)
plt.show()
#plt.savefig("随机漫步.png", bbox_inches='tight')
print('x:{}'.format(rw.x_values), 'y:{}'.format(rw.y_values), sep='\n\n')
keep_running = input("Make another walk? (y/n):")
if keep_running == 'n':
break
此为重构后的随机漫步，可以缩小fill_walk，是阅读起来更方便。