matplotlibのpick_event(クリック)時に実行される関数に引数を渡したい

matplotlibで、散布図の点をクリックした時に実行されるonpickに引数を渡したいです。
下の例だとx,yを本来は引数として渡したいのですが、方法わからずグローバル変数にして参照してます。
なにか良い方法があればご教授お願いします。

python
1import numpy as np
2import matplotlib.pyplot as plt
3import time
4
5
6def onpick(event):#散布図クリック時
7    global x,y
8    pick_ind=event.ind
9    print(x[pick_ind],y[pick_ind],pick_ind)
10    
11def plotscatter():#散布図作成
12    global x,y
13    # 乱数を生成
14    x = np.random.rand(100)
15    y = np.random.rand(100)
16
17    fig=plt.figure()
18    fig.canvas.mpl_connect('pick_event',onpick)
19    plt.scatter(x, y,picker=5)
20    plt.show()
21
22plotscatter()
23

行動規範の内容に同意します

回答1件

自己解決

自己解決しました。
lambdaを使えばできるようです。
lambda event:onpick(event,x,y)
と記述することで、マウスイベント時に渡されるイベントはeventに代入、x,yはdef plotscatter()内の値を引数として渡せました。

python
1import numpy as np
2import matplotlib.pyplot as plt
3import time
4
5
6def onpick(event,x,y):#散布図クリック時
7    print(event)
8    pick_ind=event.ind
9    print(x[pick_ind],y[pick_ind],pick_ind)
10    
11def plotscatter():
12    # 乱数を生成
13    x = np.random.rand(100)
14    y = np.random.rand(100)
15
16    fig=plt.figure()
17    event=None
18    # fig.canvas.mpl_connect('pick_event',onpick)
19    print(x)
20    fig.canvas.mpl_connect('pick_event',lambda event:onpick(event,x,y))
21
22    plt.scatter(x, y,picker=5)
23    plt.show()
24
25plotscatter()