pyplot zooming in

Use the plt.axis() function with your limits.

plt.axis([x_min, x_max, y_min, y_max])

where x_min, x_max, y_min, and y_max are the coordinate limits for both axes.