as from the topic .. sometimes happen that given a figure size for print a plot inside the label are partially cut .. as in the example reported
there is some trick in order to fit the box and the label/title all inside the figure ?
Try:
plt.savefig('fig.png', bbox_inches='tight')