2024 年高教社杯全国大学生数学建模竞赛 E 题 交通流量管控 第一问详细python代码

import pandas as pd
import numpy as np

# 1. 载入数据
direction_mapping = pd.read_excel('附件1.xlsx')  # 确保表单名称正确
vehicle_data = pd.read_csv('附件2.csv', encoding='gbk')  # 尝试正确的编码格式

# 去除列名中的空格
direction_mapping.columns = direction_mapping.columns.str.strip()
vehicle_data.columns = vehicle_data.columns.str.strip()

# 确认列名
print("方向描述列名:", direction_mapping.columns)
print("车辆数据列名:", vehicle_data.columns)

# 合并数据
merged_data = pd.merge(vehicle_data