Какая функция в питоне когда программа завершиет работу, рисунок должен остаться на экране.

Если вы говорите про то, что замыкает цикл - команда break

Если вы говорите про вывод чего-то на экран, то это будет команда print()