【python】グラフ画面をforループで更新し続けているが、グラフ画面を閉じたときもforループから抜け出したい。

Question

#やりたいこと
グラフ画面をforループで更新し続けているが、グラフ画面を閉じたときもforループから抜け出したい。

#コード

    def process_view_graph(self):
        #figureと軸初期設定
        list_hori_ax_data       =   []
        list_vergical_ax_axis   =   []
        tmp_data_graph_cnt      =   0
        fig, ax = pyplot.subplots()
        lines, = ax.plot(list_hori_ax_data, list_vergical_ax_axis)
        
        #グラフ軸設定
        for tmp_data_graph_cnt in range(100) :
            #グラフ更新
            list_hori_ax_data.append(tmp_data_graph_cnt)
            list_vergical_ax_axis.append(tmp_data_graph_cnt)     
            lines.set_data(list_hori_ax_data, list_vergical_ax_axis)
            ax.set_xlim((0, tmp_data_graph_cnt))
            ax.set_ylim((0, tmp_data_graph_cnt))   
            pyplot.pause(0.01) # これが重要！   
            

#しりたいこと
forループが終了する前にグラフ画面を閉じてもforループから抜け出す方法あるのでしょうか、何かヒントがあればご教示お願い致します。

Answer 1

以下のようにCloseイベントを捕捉すればよいです。
参考：Close Event

import matplotlib.pyplot as plt

x = []      # 描画データ
CNT = 100   # 描画回数
is_close = False

# Close イベント
def handle_close(evt):
    global is_close
    is_close = True

fig, ax = plt.subplots()
lines, = ax.plot(x, x)
fig.canvas.mpl_connect('close_event', handle_close)

ax.set_xlim((0, CNT))
ax.set_ylim((0, CNT))

for i in range(CNT):
    if is_close:
        break
    x.append(i)
    lines.set_data(x, x)

    plt.pause(0.1)

Comments

[hachimitu]

できました！ありがとうございます！助かりました。

Answer 2

forループの中で画面が開いてるかどうかをチェックして閉じてれば抜ければ。
あるいは画面を閉じるイベントでフラグセットして、ループの中でそのフラグ立ってれば抜ける、とか。