Spaces:
Build error
Build error
# 准备图数据 | |
import osmnx as ox | |
def get_graph(): | |
G = ox.load_graphml('dataset/G_new.graphml') | |
# 补充节点和边的lon,lat特征 | |
# nodes, edges = assign_edge_attr(G) | |
# 读取目前的路网 | |
# import networkx as nx | |
# G_new = nx.Graph() | |
# import tqdm | |
# print("开始读入节点") | |
# pos_location = {} | |
# for node_id, row in nodes.iterrows(): | |
# G_new.add_node(node_id, y=row['y'], x=row['x']) # 节点id,节点经纬度 | |
# pos_location[node_id] = (row['y'], row['x']) | |
# e_cnt = 0 | |
# print("开始读入边") | |
# for node_id_1, node_id_2, _ in G.edges: | |
# G_new.add_edge(node_id_1, node_id_2) # 边:节点id | |
# | |
# | |
# # 去掉自环 | |
# G_new.remove_edges_from(nx.selfloop_edges(G_new)) | |
return G | |