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 |