According to the documentation, there’s an experimental block
parameter you can pass to plt.show()
. Of course, if your version of matplotlib isn’t new enough, it won’t have this.
If you have this feature, you should be able to replace plt.show()
with plt.show(block=True)
to get your desired behavior.