将距离矩阵转化为两两之间起点加终点加距离形式

数据形式:city_name为一个列表储存城市名称

                  data是城市两两之间的od矩阵,和距离矩阵类似

import numpy as np
import pandas as pd
import openpyxl as op

# 获取城市名称
with open('city_name.txt', encoding='utf8') as f:
    city_name = f.read()
city_name = city_name.rsplit('\n')
city_name = city_name[0].split(',')
# 获取od矩阵
data=pd.read_excel('od_draw_data1.xlsx',header=None)
data = np.array(data)
data = data.tolist()
#print(data)
od_column=[]
for i in range(len(data)):

    for j in range(len(data)):
        row = []
        row.append(city_name[i]),row.append(city_name[j])
        row.append(data[i][j])
        od_column.append(row)
wb = op.Workbook()  # 创建工作簿对象
ws = wb['Sheet']  # 创建子表
for i in od_column:
    ws.append(i)
fileName='od_draw1'
wb.save(fileName + '.xlsx')

最终输出格式如下所示:

猜你喜欢

转载自blog.csdn.net/qq_45935025/article/details/121267717