matplotlib draw line from equation