右クリックでグラフの点を2回指定するとその間の点が選択され、左のダブルクリックでその範囲のデータを取得、スクロールボタンをクリックすると指定範囲のキャンセルを行うというプログラムを書きました(参考資料https://qiita.com/HajimeKawahara/items/abc24fa2216009523656)。プログラムは動くのですが、一度データを取得すると、グラフ内の別の場所を同じように指定することが出来なくなります。データを取得した後グラフの別の場所を連続して選択、データ取得をしていきたいのですがどういう風にしたらいいのでしょうか。stat=1の宣言でリセットされ、もう一度選択できると思ったのですがどうやら違うようです。どうぞ宜しくお願いします。ターミナルで動かしています。
python
1import pandas as pd 2import numpy as np 3import matplotlib.pyplot as plt 4 5x = np.array(range(20)) 6y = np.array([100,180,270,300,200,350,460,500,400,280,150,70,30,140,350,260,200,310,450,550]) 7 8def oncpaint(event): 9 global e_xdataList 10 global e_ydataList 11 global stat 12 global leftind, rightind 13 14 #event.xdataのindex情報を得る 15 ind=np.searchsorted(x, event.xdata) 16 plt.title("You clicked index="+str(ind)) 17 18 #右クリックでポイント指定1回目 19 if event.button == 3 and stat == 1: 20 leftind=ind 21 ax.plot(x[ind], y[ind], ".", color="red") 22 print( "event.button=%d, event.x=%d, event.y=%d, event.xdata=%f, \ 23 event.ydata=%f"%(event.button, event.x, event.y, event.xdata, event.ydata)) 24 stat = 2 25 #右クリックでポイント指定2回目 1回目と2回目の間の要素にマーク 26 elif event.button == 3 and stat == 2: 27 rightind=ind 28 ax.plot(x[leftind:rightind], y[leftind:rightind], ".", color="red") 29 stat = 3 30 31 #指定範囲がよければデータの保存 32 elif event.button == 1 and event.dblclick == 1 and stat == 3: 33 plt.title("Approved!") 34 print("x_value:",x[leftind:rightind+1]) 35 e_xdataList=list(x[leftind:rightind+1]) 36 e_ydataList=list(y[leftind:rightind+1]) 37 38 return e_xdataList 39 return e_ydataList 40 41 stat = 1 42 43 #指定範囲を変えたければキャンセル 44 elif event.button == 2 and stat == 3: 45 plt.title("Canceled!") 46 ax.plot(x[leftind:rightind], y[leftind:rightind], ".", color="blue") 47 stat = 1 48 fig.canvas.draw_idle() 49 50 51e_xdataList=[] 52e_ydataList=[] 53stat = 1 54 55fig=plt.figure() 56ax=fig.add_subplot(111) 57ax.scatter(x, y) 58 59fig.canvas.mpl_connect('button_press_event', oncpaint) 60plt.show() 61 62dic_event = {"x_coordinate": e_xdataList, "y_coordinate": e_ydataList} 63DicEvent={} 64for k,v in dic_event.items(): 65 DicEvent[k]=pd.Series(v) 66data=pd.DataFrame(DicEvent) 67data.to_csv("event.csv") 68 69plt.close() 70 71 72 73 74
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/06/04 19:12