会python真的可以为所欲为0(回)
355天前
这里还有人吗1(回)
540天前这里还有人吗0(回)
540天前
每天面对着电脑屏幕，敲打键盘。我所面对的并不只是代码，而是一种生活方式。0(回)
765天前到处都是羊，不想上班0(回)
845天前
鸽子
0(回)
849天前import matplotlib.pyplot as plt |
import random |
import numpy as np |
# 定义函数 |
def random_walk(n): |
x = 0 |
y = 0 |
x_list = [] |
y_list = [] |
for i in range(n): |
step = np.random.randint(1, 5) |
if step == 1: |
x += 50 |
elif step == 2: |
y += 50 |
elif step == 3: |
x -= 50 |
else: |
y -= 50 |
x_list.append(x) |
y_list.append(y) |
return x_list, y_list |
# 绘制蚂蚁随机漫步图 |
n = 30000 # 随机漫步次数 |
# 第1只蚂蚁 |
x_list, y_list = random_walk(n) |
plt.scatter(x_list, y_list, c=range(n), vmin=0, vmax=20000, s=1, cmap=plt.cm.get_cmap('PuBuGn')) |
plt.xlim((-10000, 10000)) |
plt.ylim((-10000, 10000)) |
# 第2只蚂蚁 |
x_list, y_list = random_walk(n) |
plt.scatter(x_list, y_list, c=range(n), vmin=0, vmax=20000, s=1, cmap=plt.cm.get_cmap('RdYlBu_r')) |
# 第3只蚂蚁 |
x_list, y_list = random_walk(n) |
plt.scatter(x_list, y_list, c=range(n), vmin=0, vmax=20000, s=1, cmap=plt.cm.get_cmap('YlGnBu')) |
plt.show() |
