Python求解旅行商问题

本文涉及的产品
实时计算 Flink 版,5000CU*H 3个月
检索分析服务 Elasticsearch 版,2核4GB开发者规格 1个月
大数据开发治理平台 DataWorks,不限时长
简介: 欢迎关注我的微信公众号:Python学习杂记

旅行商问题是车辆路径问题(VRP)的特例,又译为旅行推销员问题、货郎担问题,简称为TSP问题,是最基本的路线问题,该问题是在寻求单一旅行者由起点出发,通过所有给定的需求点之后,最后再回到原点的最小路径成本。

问题原型:假设有一个旅行商人要拜访n个城市,他必须选择所要走的路径,路径的限制是每个城市只能拜访一次,而且最后要回到原来出发的城市。路径的选择目标是要求得的路径路程为所有路径之中的最小值。当城市特别多的时候,求解极其复杂,由于计算机模型及算法的进步,现阶段求解方法较多。粒子算法、遗传算法、蚁群算法、神经网络等启发式方法在运筹优化求解方面都可取得较好的结果。

1.暴力法求解

暴力法:城市顺序全排列,找到最短距离。7个城市的距离矩阵如下表,现需要从第一个城市开始,最后回到第一个城市,其可能的组合有6!=720个。若城市再增加4个,其组合为10!=3628800种组合,暴力求解速度极慢,只能尝试用其他办。暴力求解需要把720条路径都计算出来,找到最短的路径即可。

import itertools
import pandas as pd
data1=pd.read_excel('distancecity.xlsx')
datanew=[0]*7
for i in range(0,7):
  datanew[i]=list(data1.iloc[i,1:8])
distance_matrix=datanew#导入数据
array = [i for i in range(1,len(distance_matrix)+1)]
pailie=list(itertools.permutations(array))
pailie_back=[list(i)+list(i)[:1] for i in pailie] #开始和结束都是一个点
pailie_back=pailie_back[:720] #取前720个,前720个为开始和结束都为第一个点的路径
def distance_all(q):
  m=0
  for i in range(len(q)-1):
      m+=distance_matrix[q[i]-1][q[i+1]-1]
  return q,m
min_distance=min([list(map(distance_all,pailie_back))[i][1] for i in range(len(pailie_back))])#搜索最小的距离路径
for i,j in map(distance_all,pailie_back):
  if j==min_distance and i[0]==1:
      print('路径:'+str([q-1 for q in i]),'总路程:'+str(j))#找到最小距离对应的路径

2.ortools求解

谷歌ortools已经搭建好了各类运输问题求解的轮子,我们只需要对数据及参数进行调整即可求解,非常方便。

from ortools.constraint_solver import routing_enums_pb2
from ortools.constraint_solver import pywrapcp
def create_data_model():
  data = {}
  data['distance_matrix'] = datanew  # 导入刚才用到的距离矩阵
  data['num_vehicles'] = 1#车辆个数,暂时设置成一辆车
  data['depot'] = 0 #起点,0是第一个为起点
  return data
def print_solution(manager, routing, solution):
  """Prints solution on console."""
  print('Objective: {} miles'.format(solution.ObjectiveValue()))
  index = routing.Start(0)
  plan_output = 'Route for vehicle 0:\n'
  route_distance = 0
  while not routing.IsEnd(index):
      plan_output += ' {} ->'.format(manager.IndexToNode(index))
      previous_index = index
      index = solution.Value(routing.NextVar(index))
      route_distance += routing.GetArcCostForVehicle(previous_index, index, 0)
  plan_output += ' {}\n'.format(manager.IndexToNode(index))
  print(plan_output)
  plan_output += 'Route distance: {}miles\n'.format(route_distance)
def main():
  """Entry point of the program."""
  # Instantiate the data problem.
  data = create_data_model()
  # Create the routing index manager.
  manager = pywrapcp.RoutingIndexManager(len(data['distance_matrix']),
                                         data['num_vehicles'], data['depot'])
  # Create Routing Model.
  routing = pywrapcp.RoutingModel(manager)
  def distance_callback(from_index, to_index):
      """Returns the distance between the two nodes."""
      # Convert from routing variable Index to distance matrix NodeIndex.
      from_node = manager.IndexToNode(from_index)
      to_node = manager.IndexToNode(to_index)
      return data['distance_matrix'][from_node][to_node]
  transit_callback_index = routing.RegisterTransitCallback(distance_callback)
  # Define cost of each arc.
  routing.SetArcCostEvaluatorOfAllVehicles(transit_callback_index)
  # Setting first solution heuristic.
  search_parameters = pywrapcp.DefaultRoutingSearchParameters()
  search_parameters.first_solution_strategy = (
      routing_enums_pb2.FirstSolutionStrategy.PATH_CHEAPEST_ARC)
  # Solve the problem.
  solution = routing.SolveWithParameters(search_parameters)
  # Print solution on console.
  if solution:
      print_solution(manager, routing, solution)
if __name__ == '__main__':
  main()

与暴力法相比,ortools速度要快很多。

3.遗传算法

遗传算法是非常经典的启发式算法,在运筹优化、经济预测、投资组合等方面都有较广泛的使用。

遗传算法其原理是:借鉴了进化生物学中的一些现象而发展起来的,这些现象包括遗传、突变、自然选择以及杂交等。遗传算法通常实现方式为一种计算机模拟。对于一个最优化问题,一定数量的候选解(称为个体)的抽象表示(称为染色体)的种群向更好的解进化。传统上,解用二进制表示(即0和1的串),但也可以用其他表示方法。进化从完全随机个体的种群开始,之后一代一代发生。在每一代中,整个种群的适应度被评价,从当前种群中随机地选择多个个体(基于它们的适应度),通过自然选择和突变产生新的生命种群,该种群在算法的下一次迭代中成为当前种群。

对于本例,先随机生成母体,由母体进行交叉、变异形成新的一代,新的一代根据其适应度(总距离)择优选择繁殖进化,最终迭代多次选择较优解。本例可用scikit-opt的算法包直接求解。

4.其他求解办法

本例除了用以上方法,还可以用经典的求解器Cplex、Gurobi等,这些求解器可进行大型案例的求解。因这两个求解器需学术申请或在校学生才可获得使用权限,若商业使用收费较高,暂无法在此展示。

相关文章
|
4天前
|
决策智能 Python
【运筹优化】(1) TSP 旅行商问题,Python + Gurobi 代码
TSP(旅行商问题)涉及寻找有向完全图中起点到所有其他点的最短回路。目标是最小化路径权重总和,保证每个节点仅访问一次。模型通过0-1决策变量表示边的存在,约束确保每个节点恰好一次作为起点和终点。为消除子圈,引入MTZ方法,添加辅助变量破坏环路。实验中,随机生成30个点,计算距离并应用MTZ模型求解,通过Gurobi库实现并展示结果。
111 0
【运筹优化】(1) TSP 旅行商问题,Python + Gurobi 代码
|
10月前
|
人工智能 Python
用python模拟地球飞越火星
当一个飞行器飞过一个行星时,如果距离较近,就会受到行星的引力影响:靠近时会逐渐加速,远离时会逐渐减速。由于能量守恒,飞行器在进入和离开行星轨道时,与行星的相对速度大小没有变化
|
11月前
|
Python
Python求解蚂蚁感冒
Python求解蚂蚁感冒
59 0
|
12月前
|
前端开发 Python
Python 实现“嫦娥奔月“小动画,一起来看看这漂亮的嫦娥吧
Python 实现“嫦娥奔月“小动画,一起来看看这漂亮的嫦娥吧
100 0
|
机器学习/深度学习 算法 测试技术
Python 中的股票预测算法
Python 中的股票预测算法
360 0
波士顿矩阵|原理+Python全流程实现
很多公司中都有着不同的产品或者是业务线,但是对于繁琐的业务来说通常我们希望根据业务的好坏进行合理的资源分配,对于这种“好坏”的判断,波士顿矩阵出现了。
波士顿矩阵|原理+Python全流程实现
python-期货指数的单位根检验
python-期货指数的单位根检验
109 0
python-分水岭-技术指标
python-分水岭-技术指标
142 0
python-CFA-计算实现理财目标需要准备的现金流期数
python-CFA-计算实现理财目标需要准备的现金流期数
61 0
|
机器学习/深度学习 Python
【l蓝桥杯】砝码称重python
【l蓝桥杯】砝码称重python
147 0