Plotting In Python Using And In If Statement