plt.savefig('output.png', dpi=300, bbox_inches='tight')

fig.tight_layout() plt.savefig('output.png', dpi=300)