|
@@ -1232,34 +1232,34 @@ class FinalDataProcessor:
|
|
|
|
|
|
# --- 合并hd_lane.csv,hd_road.csv ---
|
|
# --- 合并hd_lane.csv,hd_road.csv ---
|
|
current_file_path = os.path.abspath(__file__)
|
|
current_file_path = os.path.abspath(__file__)
|
|
- root_lane_csv_path1 = os.path.dirname(current_file_path)
|
|
|
|
- root_lane_csv_path2 = os.path.dirname(root_lane_csv_path1)
|
|
|
|
- root_lane_csv_path3 = os.path.dirname(root_lane_csv_path2)
|
|
|
|
- root_lane_csv_path4 = os.path.dirname(root_lane_csv_path3)
|
|
|
|
- lane_path = os.path.join(root_lane_csv_path4, "_internal")
|
|
|
|
- data_path = os.path.join(lane_path, "data_map")
|
|
|
|
- lane_csv_path = os.path.join(data_path, "hd_lane.csv")
|
|
|
|
- road_csv_path = os.path.join(data_path, "hd_link.csv")
|
|
|
|
- df_lane = pd.read_csv(lane_csv_path)
|
|
|
|
- column_to_read = ['link_type', 'link_coords']
|
|
|
|
-
|
|
|
|
- df_road = pd.read_csv(road_csv_path, usecols = column_to_read)
|
|
|
|
- df_road["simFrame"] = np.arange(1, len(df_road) + 1, 1)
|
|
|
|
|
|
+ # root_lane_csv_path1 = os.path.dirname(current_file_path)
|
|
|
|
+ # root_lane_csv_path2 = os.path.dirname(root_lane_csv_path1)
|
|
|
|
+ # root_lane_csv_path3 = os.path.dirname(root_lane_csv_path2)
|
|
|
|
+ # root_lane_csv_path4 = os.path.dirname(root_lane_csv_path3)
|
|
|
|
+ # lane_path = os.path.join(root_lane_csv_path4, "_internal")
|
|
|
|
+ # data_path = os.path.join(lane_path, "data_map")
|
|
|
|
+ # lane_csv_path = os.path.join(data_path, "hd_lane.csv")
|
|
|
|
+ # road_csv_path = os.path.join(data_path, "hd_link.csv")
|
|
|
|
+ # df_lane = pd.read_csv(lane_csv_path)
|
|
|
|
+ # column_to_read = ['link_type', 'link_coords']
|
|
|
|
+
|
|
|
|
+ # df_road = pd.read_csv(road_csv_path, usecols = column_to_read)
|
|
|
|
+ # df_road["simFrame"] = np.arange(1, len(df_road) + 1, 1)
|
|
# df_road = df_road.rename(columns={'link_type': 'road_type'})
|
|
# df_road = df_road.rename(columns={'link_type': 'road_type'})
|
|
|
|
|
|
|
|
|
|
- df_merged = pd.merge(
|
|
|
|
- df_merged,
|
|
|
|
- df_lane,
|
|
|
|
- on='lane_id',
|
|
|
|
- how = 'left'
|
|
|
|
- )
|
|
|
|
- df_merged = pd.merge(
|
|
|
|
- df_merged,
|
|
|
|
- df_road,
|
|
|
|
- on='simFrame',
|
|
|
|
- how = 'left'
|
|
|
|
- )
|
|
|
|
|
|
+ # df_merged = pd.merge(
|
|
|
|
+ # df_merged,
|
|
|
|
+ # df_lane,
|
|
|
|
+ # on='lane_id',
|
|
|
|
+ # how = 'left'
|
|
|
|
+ # )
|
|
|
|
+ # df_merged = pd.merge(
|
|
|
|
+ # df_merged,
|
|
|
|
+ # df_road,
|
|
|
|
+ # on='simFrame',
|
|
|
|
+ # how = 'left'
|
|
|
|
+ # )
|
|
|
|
|
|
# --- 合并 Traffic ---
|
|
# --- 合并 Traffic ---
|
|
traffic_path = self.output_dir / "Traffic.csv"
|
|
traffic_path = self.output_dir / "Traffic.csv"
|