Learn

We’ll continue building out the Vertex class. Remember, it’s responsible for knowing which other vertices are connected. These connections are the edges of our graph implementation.

A key in the Vertex instance’s edges dictionary represents a connection to that other vertex. For now, we can just set the value to be True.

grand_central = Vertex('Grand Central Station') forty_second_street = Vertex('42nd Street Station') print(grand_central.get_edges()) # [] grand_central.add_edge(forty_second_street) print(grand_central.edges) # { "42nd Street Station": True } print(grand_central.get_edges()) # ["42nd Street Station"]

Let’s add this functionality to our Vertex class!

Instructions

1.

Within Vertex, define the method .add_edge() that takes self, and vertex as arguments. The vertex argument will be the .value of another instance of Vertex.

In the body, print “Adding edge to “ + vertex.

2.

Use the vertex as a key within self.edges and set it to True.

3.

Use .add_edge() to assign forty_second_street.value as an edge of grand_central.

Folder Icon

Sign up to start coding

By signing up for Codecademy, you agree to Codecademy's Terms of Service & Privacy Policy.

Already have an account?