[matplotlib] plt warning (More than 20 figures have been opened.)

matplotlib plt.subplots warning 삭제 

 

for문과 matplotlib를 사용해서 이미지를 그릴 때, 다음과 같은 warning이 나오는 경우가 있다.

 

RuntimeWarning: More than 20 figures have been opened. Figures created through the pyplot interface (`matplotlib.pyplot.figure`) are retained until explicitly closed and may consume too much memory. (To control this warning, see the rcParam `figure.max_open_warning`).

 

이럴때는 내가 열어준 fig를 close()해주지 않아서 생기는 문제로, 이미지를 save 후 다음 iteration으로 돌기 직전에 아래 코드를 삽입해주면 뜨지 않는다.

plt.close(fig)