matplotlibのグラフ上でクリックしたポイントのx座標をリストに保存したい

Question

matplotlibのscatter plot上で選択したポイントのx座標を保存したいです。自分で書いたプログラムをターミナル上で動かすと、毎回保存したx座標が上書きされているようで最後に選択したポイントの値しか記録できません。選択したすべての座標を取得するにはどうしたらいいのでしょうか。ご教示いただければ幸いです。

import pandas as pd

import matplotlib.pyplot as plt
import random
import numpy as np



x1=[random.random() for i in range(1000)]
y1=[random.randint(0, 5000) for i in range(1000)]

def onclick(event):
    event_xdata_list=[]
    event_xdata_list.append(event.xdata)
    data=pd.DataFrame(event_xdata_list)
    data.to_csv("event_x.csv")

    print( "event.button=%d,  event.x=%d, event.y=%d, event.xdata=%f, \
    event.ydata=%f"%(event.button, event.x, event.y, event.xdata, event.ydata))
    
    
fig=plt.figure(figsize=(16.0,8.0))
ax=fig.add_subplot(111)

ax.scatter(x1, y1, c="blue", s=3, label="rpm")

ax.set_xlabel("time(s)")
ax.set_ylabel("rpm")


ax.set_xlim([0,1])
ax.set_ylim([0, 5000])

fig.canvas.mpl_connect('button_press_event', onclick)

plt.show()
plt.close()

Answer 1

onclick関数の中にデータ保存処理が在るため、繰り返し上書きされていたのだと思います。
関数の中に書いてあるデータ保存処理を外に出しました。

import pandas as pd
import matplotlib.pyplot as plt
import random
import numpy as np

x1=[random.random() for i in range(1000)]
y1=[random.randint(0, 5000) for i in range(1000)]

def onclick(event):
    global event_xdata_list
    print("event.button=%d,  event.x=%d, event.y=%d, event.xdata=%f, \
    event.ydata=%f"%(event.button, event.x, event.y, event.xdata, event.ydata))
    event_xdata_list.append(event.xdata)

fig=plt.figure(figsize=(16.0,8.0))
ax=fig.add_subplot(111)

ax.scatter(x1, y1, c="blue", s=3, label="rpm")
ax.set_xlabel("time(s)")
ax.set_ylabel("rpm")
ax.set_xlim([0,1])
ax.set_ylim([0, 5000])

event_xdata_list = list()
fig.canvas.mpl_connect('button_press_event', onclick)
plt.show()
print("positions are ",event_xdata_list)
data=pd.DataFrame(event_xdata_list)
data.to_csv("event_x.csv")
plt.close()

Comments

[Yumame]

どうもありがとうございました！！やりたかったことが出来ました。コードを書く順番も大変勉強になりました。global変数は聞いたことはあったのですが使ったこともなくよく分かっていなかったのですが、これをみて使い方が理解できました！感謝です。