how to create a graph in python