def DrawGraph(G, centers):
	pos = nx.spring_layout(G)
	color_map = ['blue'] * len(G.nodes())
 	#all the center nodes are marked with 'red'
	for c in centers:
		color_map[c] = 'red'
	nx.draw(G, pos, node_color = color_map, with_labels = True)  #with_labels=true is to show the node number in the output graph
	edge_labels = nx.get_edge_attributes(G, 'length')
	nx.draw_networkx_edge_labels(G, pos, edge_labels = edge_labels, font_size = 11) #prints weight on all the edges
	


#main function
if __name__ == "__main__":
	G,n = CreateGraph()
	centers = k_centers(G, n)
	DrawGraph(G, centers)
	plt.show()