How to save plot after the plot is zoomed

Question

Last few days I have been facing a problem in saving the matplotlib figure, three days before the code was working fine and saves the plot with changes made through the code, but now the changes (zoomed plot) is not saving rather it saves as it is when the plot is shown any changes made after does not reflects using the save command, don't know why?

    ax = pd.rolling_mean(dataToPlot_plot[startTime:endTime][['plotValue']],mar).plot(linestyle='-', linewidth=3,  markersize=9, color='#FECB00')
            ax.legend().set_visible(False)
    plt.show()#showing the plot
    fig = ax.get_figure()
    fig.set_size_inches(12, 6)#setting the size of the plot, to fix in the PDF file
    fig.savefig('graph1.jpg')#saving the plot

even if I call a function, the new changed plot is not saved...

    def callmeto_plot()
            ax = pd.rolling_mean(dataToPlot_plot[startTime:endTime][['plotValue']],mar).plot(linestyle='-', linewidth=3,  markersize=9, color='#FECB00')
            ax.legend().set_visible(False)
            plt.show()#showing the plot
            fig = ax.get_figure()

            return fig

    fig = callmeto_plot()
    fig.set_size_inches(12, 6)
    fig.savefig('graph1.jpg')

How do I save the plot (zoomed one with changes) through code?
Note: I have noticed that the plot window appearance is also changed,
1 Plot window appearance before
2 Plot window appearance now
all the plot window configuration buttons are shifted from bottom to top, does this change affects only in the plot or affects the coding as well?. Please help me to fix this...
Thanks in advance.

Answer 1

You could move all the plot changing code above the plt.show() and then save the figure manually from the pop-up window. The right most icon at the bottom left saves the current figure as is currently displayed

Even better you could use plt.axis([xStart, xFinish, yBottom, yTop]) or ax.set_xlim(), ax.set_ylim