実現したいこと
matplotlibでプロットした折れ線グラフでのpickイベントについてです。グラフ内に複数の線をプロットし、各線をクリックしたときに、クリックされた線のデータのインデックスを取得したいです。例えば、グラフ内の一つの線をクリックしたときは、その線のインデックス(プロットしたときの順番)を1つ表示し、線が重なっていたときは重なっている複数の線のインデックスを表示したいです。
問題
pickイベント自体は動いているようですが、複数の線が重なる場合、そのうちの1つの線に対応するインデックスだけが複数回表示されてしまいます。例えば、重なっている線1(インデックス:1)、線2(インデックス:2)をクリックすると、pickイベント関数が一度だけ呼び出され、evnet.ind=[1, 2]となるのが理想ですが、そうではなく、重なった線と同じ回数pickイベント関数が呼び出されて、その全てでevent.ind=[1]となります。このとき、evnet.indは必ず1つの要素しか持ちません。
scatter()で同じように散布図内でのpickイベントを実装した際には、scatter()をデータ数と同じ回数実行して複数のデータをプロットすると、各点が異なるコレクションの点として扱われてしまい、重なる点に対するクリックで、クリックされた複数点のインデックスを一度に取得することができなかった経験がありました。なので、以下のソースコードの様に、複数の線を一度のplotで追加しましたが、うまくいきませんでした。そもそも表示された各線は色が違い、event.artistからも確認したところ、異なるコレクション(Line2D)として扱われているようです。同じコレクションとして折れ線グラフをプロットすることはできないのでしょうか.
該当のソースコード
Python
1import matplotlib.pyplot as plt 2import numpy as np 3 4def pickEvent(event): 5 print(f'artist:{event.artist}') 6 global cnt 7 print(f'pick event cnt:{cnt}') 8 cnt += 1 9 print(f'ind:{event.ind}') 10 11cnt = 0 12N = 10 13M = 10 14x = np.arange(M) 15y = np.array([np.random.rand(M) for i in range(N)]) 16 17fig = plt.figure(figsize=(4, 4)) 18ax = fig.add_subplot(111) 19ax.plot(x, y, picker=True, pickradius=1.0) 20fig.canvas.mpl_connect('pick_event', pickEvent) 21plt.show() 22
あなたの回答
tips
プレビュー