В конце последней функции, которую я вызываю в одной из моих программ, у меня есть следующий код для построения простой цветной диаграммы.

plt.figure()
plt.pcolormesh(X,Y,Z)
plt.colorbar()
plt.show()

После слов я возвращаюсь к основному, и моя программа завершена. График отображается должным образом, однако, когда я закрываю его с помощью кнопки x в углу (в Ubuntu), моя программа нет конец. Он просто висит там с запущенным процессом. Как я могу это исправить?