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