I'm trying to implement a method that returns the edges of a graph, represented by an adjacency list/dictionary.
So to iterate through the dictionary, first I iterated through the keys, then through every value stored in the corresponding key. Inside the nested for-loop, I had a condition where, if a particular edge, say (a,b) is not in the set of edges, then add it to the set -- pass otherwise. On my first run, the method took in edges that are the same -- that is, in the set of edges, there are (a,b) and (b,a).
class Graph():
def __init__(self, grph={}):
self.graph = grph
def get_vertices(self):
for keys in self.graph:
yield keys
def get_edges(self):
edges = set()
for key in self.graph:
for adj_node in self.graph[key]:
if (key, adj_node) not in edges:
edge = (key, adj_node)
edges.add(edge)
else:
pass
return edges
def main():
graph1 = {
'A': ['B','C','D'],
'B': ['A','E'],
'C': ['A', 'D'],
'D': ['A', 'C'],
'E': ['B'],
}
graph_one = Graph(graph1)
print(list(graph_one.get_vertices()))
print(graph_one.get_edges())
if __name__ =='__main__':
main()
the output is:
{('A','B'),('D','A'),('B','A'),('B','E'),('A','D'),('D','C'),('E','B'),('C','D'),('A','C'),('C','A')}
So what I did was that, I just changed the if statement:
"if (adj_node, key) not in edges:"
def get_edges(self):
edges = set()
for key in self.graph:
for adj_node in self.graph[key]:
if (adj_node, key) not in edges:
edge = (key, adj_node)
edges.add(edge)
else:
pass
return edges
Now the output was:
{('C','D'),('A','B'),('E','B'),('A','C'),('A','D')}
Im very curious as to why it is so, and I'd be so thankful if you guys could explain it to me. Thanks in advance!
sortthe data or use an helper set with sorted data to avoid insertion. What's the type of key and adj_node?setwill only holds one instance of each value, and you usedifto check opposite direction's tuple, so in effect both directions' duplicates are avoided.