Browse Source

alter the location of ids saving codes

master
김선중 1 year ago
parent
commit
937d0a9df7
1 changed files with 10 additions and 10 deletions
  1. +10
    -10
      Scripts/preprocess_daily.py

+ 10
- 10
Scripts/preprocess_daily.py View File

@ -191,6 +191,16 @@ class DailyPreprocessor():
self.child_ids = sorted(self.inter_node[self.inter_node.inter_type=='child'].node_id.unique()) self.child_ids = sorted(self.inter_node[self.inter_node.inter_type=='child'].node_id.unique())
self.uturn_ids = sorted(self.uturn.child_id.unique()) self.uturn_ids = sorted(self.uturn.child_id.unique())
self.coord_ids = sorted(self.coord.child_id.unique()) self.coord_ids = sorted(self.coord.child_id.unique())
# ids
ids = {'node_ids' : self.node_ids,
'parent_ids': self.parent_ids,
'child_ids' : self.child_ids,
'uturn_ids' : self.uturn_ids,
'coord_ids' : self.coord_ids,
'inter_nos' : self.inter_nos}
with open(os.path.join(self.path_intermediates, 'ids.json'), 'w') as file:
json.dump(ids, file)
print("1-5. 주요 객체 (리스트, 딕셔너리)들을 저장했습니다.") print("1-5. 주요 객체 (리스트, 딕셔너리)들을 저장했습니다.")
# 2. 중간산출물 만들기 # 2. 중간산출물 만들기
@ -387,16 +397,6 @@ class DailyPreprocessor():
: inter_no, phase_no, ring_type, move_no, inc_dire, out_dire, inc_angle, out_angle, inc_edge_id, out_edge_id, node_id : inter_no, phase_no, ring_type, move_no, inc_dire, out_dire, inc_angle, out_angle, inc_edge_id, out_edge_id, node_id
''' '''
# ids
ids = {'node_ids' : self.node_ids,
'parent_ids': self.parent_ids,
'child_ids' : self.child_ids,
'uturn_ids' : self.uturn_ids,
'coord_ids' : self.coord_ids,
'inter_nos' : self.inter_nos}
with open(os.path.join(self.path_intermediates, 'ids.json'), 'w') as file:
json.dump(ids, file)
ch2pa = {} # child to parent ch2pa = {} # child to parent
for child_id in self.child_ids: for child_id in self.child_ids:
parent_no = self.inter_node[self.inter_node.node_id==child_id].inter_no.iloc[0] parent_no = self.inter_node[self.inter_node.node_id==child_id].inter_no.iloc[0]

Loading…
Cancel
Save