18 lines
643 B
Python
18 lines
643 B
Python
|
import numpy as np
|
|||
|
|
|
|||
|
|
|
|||
|
|
def convert_to_partition(yanlinks):
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
将数据转化为分区的结果
|
|||
|
|
yanlinks: 一个NumPy数组，第五列是路段编号，第12列是分区编号
|
|||
|
|
返回值: 分区的结果，是一个列表，列表中的每个元素是一个集合，表示一个分区
|
|||
|
|
"""
|
|||
|
|
partition = []
|
|||
|
|
# 获取所有的分区编号
|
|||
|
|
zone_ids = np.unique(yanlinks[:, 11])
|
|||
|
|
# 对于每个分区编号，找出对应的路段编号，添加到partition中
|
|||
|
|
for zone_id in zone_ids:
|
|||
|
|
segment_ids = set(yanlinks[yanlinks[:, 11] == zone_id, 4])
|
|||
|
|
partition.append(segment_ids)
|
|||
|
|
return partition
|