python 实现蚁群算法（simpy带绘图）

这里使用了蚁群算法求解了旅行商问题，同时结合了simpy来绘图

选择下一个食物的函数为：

probability[i] = pheromone[self.now][self.not_to_foods[i]] ** pheromone_w + (1 / distance[self.now][self.not_to_foods[i]]) ** distance_w

该条路概率权重=该点信息素^信息素权重*（1/路径长）^路径权重/总和，这里后面会用random.choices就不用除了

信息素更新为：

new_pheromone[self.route[i]][self.route[i + 1]] += pheromone_Q / self.sum_len

pheromone = np.add((1 - volatilize) * pheromone, new_pheromone)

总信息素会挥发volatilize比例，然后所有蚂蚁路径每一段加上固定释放的信息量/路径总长度

话不多说，都在代码注释里了

import random
import simpy
import matplotlib.pyplot as plt
import numpy as np# 在这里我们尝试用蚁群算法求解旅行商问题，简而言之，就是寻找一条链接所有点的最短路径最后要回到初始点
# 我想绘制一下每只蚂蚁的行走过程，所以使用上次的simpy包来模拟一下，
# 这个simpy包主要是为了更好的可视化模拟训练过程，你也可以给每只蚂蚁单独线程代替，
# 或者干脆不用管过程，只用一轮一轮的上蚂蚁，在信息素部分做点处理，光看结果也可以。总之大多要比simpy快class Ant(object):  # 蚂蚁类# 说一下训练过程，有很多只蚂蚁，从随机位置出发，# 每次遵从信息素的指引，选择下一个城市，直到没有可选城市，回到初始位置然后根据路径总长度在路径上散播信息素def __init__(self):  # 蚂蚁构造函数# 一只蚂蚁，只需要保存走过的食物就可以，# 这里我们为了以后整信息素时不用遍历，顺便保存一下走过的长度self.route = []  # 到达食物路径，一个例子是：[2,1,3,2]self.not_to_foods = []  # 未到达的食物点self.sum_len = 0  # 走过路径总长度self.now = 0  # 当前所在位置self.next_food = 0  # 下一个所在位置self.Initial_Position(random.randint(0,foods_n-1))  # 设定初始位置# 绘图参数self.now_position = [0, 0]  # 起始到终点的向量self.proportion = env.now  # 走的时间，画图就是找到起始食物点，然后加上（起始和终点的向量乘以（当前时间-出发时间）/路长）就是蚂蚁位置，def Initial_Position(self, n):  # 选择初始位置'self.now = n  # 当前所在的食物self.next_food = n  # 下一个要到的食物self.route = []self.not_to_foods = [i for i in range(foods_n)]  # 未到达的食物点self.sum_len = 0self.route.append(n)  # 添加到路径self.not_to_foods.remove(n)  # 初始点不会在未到的食物点def Set_Now(self, next_food):  # 到下一个点self.next_food = next_food  # 下一个要到的食物self.route.append(next_food)  # 添加到路径self.sum_len += distance[self.now][next_food]  # 总路径增加if next_food in self.not_to_foods:self.not_to_foods.remove(next_food)  # 移出未到的食物点self.now_position = foods[next_food] - foods[self.now]  # 走的方向self.proportion = env.now  # 出发时间def Next_Food(self):  # 根据信息素和路径寻找下一个食物len_not_to_foods = len(self.not_to_foods)  # 没选择的食物个数probability = np.zeros(len_not_to_foods)  # 选择食物的概率for i in range(len_not_to_foods):# 选一个路径的概率是该点信息素^信息素权重*（1/路径长）^路径权重/总和，这里后面会用random.choices就不用除了probability[i] = pheromone[self.now][self.not_to_foods[i]] ** pheromone_w + (1 / distance[self.now][self.not_to_foods[i]]) ** distance_wnext_food = random.choices(self.not_to_foods, probability)  # 根据权重选择下一个点return next_fooddef Change_Pheromone(self):  # 更新信息素矩阵，先添加到一个临时变量中，信息素矩每隔一段时间后自己会挥发并将这个变量添加进去。for i in range(foods_n):new_pheromone[self.route[i]][self.route[i + 1]] += pheromone_Q / self.sum_lennew_pheromone[self.route[i + 1]][self.route[i]] = new_pheromone[self.route[i]][self.route[i + 1]]def run(self):  # 蚂蚁出动global min_distance, min_routewhile True:while self.not_to_foods:  # 未空还能找食物next_food = self.Next_Food()[0]self.Set_Now(next_food)  # 根据信息素和路径去下一个食物位置yield env.timeout(distance[self.now][next_food])  # 等待到达终点self.now = next_food  # 到达位置# 没有食物可以去了next_food = self.route[0]self.Set_Now(next_food)  # 去初始点yield env.timeout(distance[self.now][next_food])  # 等待到达终点self.now = next_food  # 到达位置self.Change_Pheromone()  # 到了初始点之后，更新信息素if min_distance > self.sum_len:min_distance = self.sum_lenmin_route = self.routeself.Initial_Position(random.randint(0,foods_n-1))  # 重新初始化，再来一轮def Change_Pheromone(env):global pheromone, new_pheromonewhile True:# 每隔一段时间，挥发，更新信息素，# 我想要平均走完一波蚂蚁更新一波，那么，我大概需要等待平均总路长的时间（因为假设蚂蚁1m/s）这里就用(foods_n+1)/2,两点间平均距离大概是0.5yield env.timeout((foods_n + 1) / 2)pheromone = np.add((1 - volatilize) * pheromone, new_pheromone)new_pheromone = np.zeros((foods_n, foods_n))# 初始化函数，初始化蚁群，食物，信息素矩阵，
def Initialization(ants_n, foods_n, dimension):''':param ants_n:  蚁群大小:param foods_n: 食物数目:param dimension: 维度:return: 初始化蚁群，食物，信息素矩阵，距离矩阵'''ants = [Ant() for _ in range(ants_n)]  # 蚁群np.random.seed(0)  # 随机数种子，让生成的位置一致,你可以删去foods = np.random.rand(foods_n, dimension)  # 所有食物位置，支持高维，不过绘图只有二维（python三维图太卡了），只测试二维pheromone = np.zeros((foods_n, foods_n))+2  # 信息素矩阵初始为路径长度平均数的倒数之类的distance = np.zeros((foods_n, foods_n))  # 距离矩阵for i in range(foods_n):  # 计算欧氏距离for j in range(foods_n):if i > j:  # 以前算过了，抄过来distance[i][j] = distance[j][i]continueif i == j:  # 一样的位置就是0，不算了continuedistance[i][j] = np.sqrt(np.sum((foods[i] - foods[j]) ** 2))return ants, foods, pheromone, distancedef plt_Refresh():  # 绘图while True:plt.clf()  # 清屏plt.xlim(0, 1)plt.ylim(-0.1, 1)# 绘图plt.text(0.05, -0.05, "now_time = " + str(env.now))for i in range(foods_n):  # 所有的食物点for j in range(i):plt.plot([foods[i][0],foods[j][0]],[foods[i][1],foods[j][1]],linewidth=pheromone[i][j]/100)plt.scatter(foods[i][0], foods[i][1], 100)plt.text(foods[i][0], foods[i][1], i)for i in range(ants_n):  # 所有的蚂蚁点# 到起始食物点，然后加上（起始和终点的向量乘以（当前时间-出发时间）/路长）就是蚂蚁位置，if ants[i].now == ants[i].next_food:position = foods[ants[i].now]else:position = foods[ants[i].now] + ants[i].now_position * (env.now - ants[i].proportion) / \distance[ants[i].now][ants[i].next_food]plt.scatter(position[0], position[1])plt.text(position[0], position[1], i)# 刷新图形plt.draw()plt.pause(time_particles)yield env.timeout(time_particles)def Firing(env):  # env启动env.process(Change_Pheromone(env))  # 启动信息素矩阵env.process(plt_Refresh())  # 启动绘图for i in range(ants_n):  # 启动所有蚂蚁env.process(ants[i].run())ants_n = 15  # 蚂蚁数目，一般1.5*foods_n
foods_n = 10  # 食物数目
dimension = 2  # 维度，建议2，因为绘图只搞了2
min_distance = float("inf")  # 最小距离
min_route = []  # 最小路径
# 一些超参数
pheromone_w = 1  # 信息素权重
distance_w = 6  # 距离权重
volatilize = 0.5  # 信息素挥发比例
pheromone_Q = 10  # 总信息素
run_time = 100  # 模拟时间
time_particles = 0.05  # 绘图间隔
new_pheromone = np.zeros((foods_n, foods_n))
# 绘图
plt.figure()
plt.pause(10)    # 方便我录屏，等10s再出画面你们要删去
env = simpy.Environment()  # 设置环境并启动模拟
ants, foods, pheromone, distance = Initialization(ants_n, foods_n, dimension)  # 初始化,通常，蚁群数目是食物数目1.5倍
print(foods)
Firing(env)  # 开火，启动（添加要模拟的函数）
env.run(until=run_time)  # 运行模拟print(min_route,min_distance)
plt.show()  # 遍历完成后不消失

自己用建议删去 plt.pause(10)这一行，同时，可以调大time_particles，

这是结果：

视频审核还没通过，通过了就放出来