jupyter不显示直接保存图片

在Jupyter中，画很多图时显示在代码下方会占很多内存进而卡顿，有时我们只需要保存它而不想显示怎么办呢？在bing和百度用中文搜都没有查到很好的解决办法，有的用matplotlib.use(‘Agg’), 有的用魔术命令%doctest_mode，但是某些情况下还是没用且不直接。

实际上直接在画完图保存之后加上

plt.close(fig)

就行了。
示例代码

import matplotlib.pyplot as plt
fig, ax = plt.subplots( nrows=1, ncols=1 ) # create figure & 1 axis
ax.plot([0,1,2], [10,20,3])
fig.savefig('path/to/save/image/to.png') # save the figure to file
plt.close(fig) # close the figure