Graph—Undirected graphs with self loops¶
Overview¶
-
class
Graph
(data=None, **attr)[source]¶ Base class for undirected graphs.
A Graph stores nodes and edges with optional data, or attributes.
Graphs hold undirected edges. Self loops are allowed but multiple (parallel) edges are not.
Nodes can be arbitrary (hashable) Python objects with optional key/value attributes. By convention
None
is not used as a node.Edges are represented as links between nodes with optional key/value attributes.
Parameters: - data (input graph) – Data to initialize graph. If data=None (default) an empty graph is created. The data can be any format that is supported by the to_networkx_graph() function, currently including edge list, dict of dicts, dict of lists, NetworkX graph, NumPy matrix or 2d ndarray, SciPy sparse matrix, or PyGraphviz graph.
- attr (keyword arguments, optional (default= no attributes)) – Attributes to add to graph as key=value pairs.
See also
Examples
Create an empty graph structure (a “null graph”) with no nodes and no edges.
>>> G = nx.Graph()
G can be grown in several ways.
Nodes:
Add one node at a time:
>>> G.add_node(1)
Add the nodes from any container (a list, dict, set or even the lines from a file or the nodes from another graph).
>>> G.add_nodes_from([2, 3]) >>> G.add_nodes_from(range(100, 110)) >>> H = nx.path_graph(10) >>> G.add_nodes_from(H)
In addition to strings and integers any hashable Python object (except None) can represent a node, e.g. a customized node object, or even another Graph.
>>> G.add_node(H)
Edges:
G can also be grown by adding edges.
Add one edge,
>>> G.add_edge(1, 2)
a list of edges,
>>> G.add_edges_from([(1, 2), (1, 3)])
or a collection of edges,
>>> G.add_edges_from(H.edges)
If some edges connect nodes not yet in the graph, the nodes are added automatically. There are no errors when adding nodes or edges that already exist.
Attributes:
Each graph, node, and edge can hold key/value attribute pairs in an associated attribute dictionary (the keys must be hashable). By default these are empty, but can be added or changed using add_edge, add_node or direct manipulation of the attribute dictionaries named graph, node and edge respectively.
>>> G = nx.Graph(day="Friday") >>> G.graph {'day': 'Friday'}
Add node attributes using add_node(), add_nodes_from() or G.nodes
>>> G.add_node(1, time='5pm') >>> G.add_nodes_from([3], time='2pm') >>> G.nodes[1] {'time': '5pm'} >>> G.nodes[1]['room'] = 714 # node must exist already to use G.nodes >>> del G.nodes[1]['room'] # remove attribute >>> list(G.nodes(data=True)) [(1, {'time': '5pm'}), (3, {'time': '2pm'})]
Add edge attributes using add_edge(), add_edges_from(), subscript notation, or G.edges.
>>> G.add_edge(1, 2, weight=4.7 ) >>> G.add_edges_from([(3, 4), (4, 5)], color='red') >>> G.add_edges_from([(1, 2, {'color': 'blue'}), (2, 3, {'weight': 8})]) >>> G[1][2]['weight'] = 4.7 >>> G.edges[1, 2]['weight'] = 4
Warning: we protect the graph data structure by making
G.edges[1, 2]
a read-only dict-like structure. Use 2 sets of brackets to add/change data attributes. (For multigraphs:MG.edges[u, v, key][name] = value
).Shortcuts:
Many common graph features allow python syntax to speed reporting.
>>> 1 in G # check if node in graph True >>> [n for n in G if n < 3] # iterate through nodes [1, 2] >>> len(G) # number of nodes in graph 5
Often the best way to traverse all edges of a graph is via the neighbors. The neighbors are reported as an adjacency-dict
G.adj
or asG.adjacency()
>>> for n, nbrsdict in G.adjacency(): ... for nbr, eattr in nbrsdict.items(): ... if 'weight' in eattr: ... # Do something useful with the edges ... pass
But the edges() method is often more convenient:
>>> for u, v, weight in G.edges.data('weight'): ... if weight is not None: ... # Do something useful with the edges ... pass
Reporting:
Simple graph information is obtained using object-attributes and methods. Reporting typically provides views instead of containers to reduce memory usage. The views update as the graph is updated similarly to dict-views. The objects
nodes, `edges
andadj
provide access to data attributes via lookup (e.g.nodes[n], `edges[u, v]
,adj[u][v]
) and iteration (e.g.nodes.items()
,nodes.data('color')
,nodes.data('color', default='blue')
and similarly foredges
) Views exist fornodes
,edges
,neighbors()
/adj
anddegree
.For details on these and other miscellaneous methods, see below.
Subclasses (Advanced):
The Graph class uses a dict-of-dict-of-dict data structure. The outer dict (node_dict) holds adjacency information keyed by node. The next dict (adjlist_dict) represents the adjacency information and holds edge data keyed by neighbor. The inner dict (edge_attr_dict) represents the edge data and holds edge attribute values keyed by attribute names.
Each of these three dicts can be replaced in a subclass by a user defined dict-like object. In general, the dict-like features should be maintained but extra features can be added. To replace one of the dicts create a new graph class by changing the class(!) variable holding the factory for that dict-like structure. The variable names are node_dict_factory, adjlist_inner_dict_factory, adjlist_outer_dict_factory, and edge_attr_dict_factory.
- node_dict_factory : function, (default: dict)
- Factory function to be used to create the dict containing node attributes, keyed by node id. It should require no arguments and return a dict-like object
- adjlist_outer_dict_factory : function, (default: dict)
- Factory function to be used to create the outer-most dict in the data structure that holds adjacency info keyed by node. It should require no arguments and return a dict-like object.
- adjlist_inner_dict_factory : function, (default: dict)
- Factory function to be used to create the adjacency list dict which holds edge data keyed by neighbor. It should require no arguments and return a dict-like object
- edge_attr_dict_factory : function, (default: dict)
- Factory function to be used to create the edge attribute dict which holds attrbute values keyed by attribute name. It should require no arguments and return a dict-like object.
Examples
Create a low memory graph class that effectively disallows edge attributes by using a single attribute dict for all edges. This reduces the memory used, but you lose edge attributes.
>>> class ThinGraph(nx.Graph): ... all_edge_dict = {'weight': 1} ... def single_edge_dict(self): ... return self.all_edge_dict ... edge_attr_dict_factory = single_edge_dict >>> G = ThinGraph() >>> G.add_edge(2, 1) >>> G[2][1] {'weight': 1} >>> G.add_edge(2, 2) >>> G[2][1] is G[2][2] True
Please see
ordered
for more examples of creating graph subclasses by overwriting the base classdict
with a dictionary-like object.
Methods¶
Adding and removing nodes and edges¶
Graph.__init__ ([data]) |
Initialize a graph with edges, name, or graph attributes. |
Graph.add_node (n, **attr) |
Add a single node n and update node attributes. |
Graph.add_nodes_from (nodes, **attr) |
Add multiple nodes. |
Graph.remove_node (n) |
Remove node n. |
Graph.remove_nodes_from (nodes) |
Remove multiple nodes. |
Graph.add_edge (u, v, **attr) |
Add an edge between u and v. |
Graph.add_edges_from (ebunch, **attr) |
Add all the edges in ebunch. |
Graph.add_weighted_edges_from (ebunch[, weight]) |
Add all the weighted edges in ebunch with specified weights. |
Graph.remove_edge (u, v) |
Remove the edge between u and v. |
Graph.remove_edges_from (ebunch) |
Remove all edges specified in ebunch. |
Graph.clear () |
Remove all nodes and edges from the graph. |
Reporting nodes edges and neighbors¶
Graph.nodes |
A NodeView of the Graph as G.nodes or G.nodes(). |
Graph.__iter__ () |
Iterate over the nodes. |
Graph.has_node (n) |
Return True if the graph contains the node n. |
Graph.__contains__ (n) |
Return True if n is a node, False otherwise. |
Graph.edges |
An EdgeView of the Graph as G.edges or G.edges(). |
Graph.has_edge (u, v) |
Return True if the edge (u, v) is in the graph. |
Graph.get_edge_data (u, v[, default]) |
Return the attribute dictionary associated with edge (u, v). |
Graph.neighbors (n) |
Return an iterator over all neighbors of node n. |
Graph.adj |
Graph adjacency object holding the neighbors of each node. |
Graph.__getitem__ (n) |
Return a dict of neighbors of node n. |
Graph.adjacency () |
Return an iterator over (node, adjacency dict) tuples for all nodes. |
Graph.nbunch_iter ([nbunch]) |
Return an iterator over nodes contained in nbunch that are also in the graph. |
Counting nodes edges and neighbors¶
Graph.order () |
Return the number of nodes in the graph. |
Graph.number_of_nodes () |
Return the number of nodes in the graph. |
Graph.__len__ () |
Return the number of nodes. |
Graph.degree |
A DegreeView for the Graph as G.degree or G.degree(). |
Graph.size ([weight]) |
Return the number of edges or total of all edge weights. |
Graph.number_of_edges ([u, v]) |
Return the number of edges between two nodes. |
Making copies and subgraphs¶
Graph.copy ([as_view]) |
Return a copy of the graph. |
Graph.to_undirected ([as_view]) |
Return an undirected copy of the graph. |
Graph.to_directed ([as_view]) |
Return a directed representation of the graph. |
Graph.subgraph (nodes) |
Return a SubGraph view of the subgraph induced on nodes . |
Graph.edge_subgraph (edges) |
Returns the subgraph induced by the specified edges. |
Graph.fresh_copy () |
Return a fresh copy graph with the same data structure. |