数据形式: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')
最终输出格式如下所示: