Python Code For Plotting Graph