Python Matplotlib Set Axis Limits Python Interview