import matplotlib.pyplot as plt
from random_walk import RandomWalk
# Keep making random walks, as long as the program is active
while True:
rw = RandomWalk()
rw.fill_walk()
point_numbers = list(range(rw.num_points))
plt.scatter(rw.x_values, rw.y_values, c=point_numbers, cmap=plt.cm.Blues, edgecolors='none', s=10)
# Emphasize the first and last points.
plt.scatter(0, 0, c='green', edgecolors='none', s=50)
plt.scatter(rw.x_values[-1], rw.y_values[-1], c='red', s=50)
# Remove the axes.
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
import matplotlib.pyplot as plt
from random_walk import RandomWalk
# 解决不能显示中文
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
while True:
# 创建一个RandomWalk实例,并绘制
rw=RandomWalk()
rw.fill_walk()
points_numbers=list(range(rw.num_points))
# 设置图表标题,并给坐标轴加上标签
plt.title("随机漫步图", fontsize=24)
plt.scatter(rw.x_values,rw.y_values,c=points_numbers,cmap=plt.cm.Blues,edgecolors='none',s=15)
# 隐藏坐标轴
plt.gca().get_xaxis().set_visible(False)
plt.gca().get_yaxis().set_visible(False)
#plt.show()
plt.savefig('sjmb.png',bbox_inches='tight')
keep_running=input("是否要再模拟一次随机漫步?(y/n):")
if keep_running=='n':
break