osmnx compat

momepy/utils.py

@@ -90,6 +90,9 @@ def _generate_primal(graph, gdf_network, fields, multigraph, oneway_column=None)
         else:
             graph.add_edge(first, last, **attributes)
 
+        node_attrs = {node: {"x": node[0], "y": node[1]} for node in graph.nodes}
+        nx.set_node_attributes(graph, node_attrs)
+
 
 def _generate_dual(graph, gdf_network, fields, angles, multigraph, angle):
     """Generate a dual graph. Helper for ``gdf_to_nx``."""
@@ -150,6 +153,7 @@ def gdf_to_nx(
     angles=True,
     angle="angle",
     oneway_column=None,
+    integer_labels=False,
 ):
     """
     Convert a LineString GeoDataFrame to a ``networkx.MultiGraph`` or other
@@ -188,6 +192,10 @@ def gdf_to_nx(
         path traversal by specifying the boolean column in the GeoDataFrame. Note,
         that the reverse conversion ``nx_to_gdf(gdf_to_nx(gdf, directed=True,
         oneway_column="oneway"))`` will contain additional duplicated geometries.
+    integer_labels : bool, default False
+        Convert node labels to integers. By default, node labels are tuples with (x, y)
+        coordinates. Set to True to encode them as integers. Note that the x, and y
+        coordinates are always preserved as node attributes.
 
     Returns
     -------
@@ -273,6 +281,9 @@ def gdf_to_nx(
             f"Approach '{approach}' is not supported. Use 'primal' or 'dual'."
         )
 
+    if integer_labels:
+        net = nx.convert_node_labels_to_integers(net)
+
     return net