クリックによりxy平面上の座標点を保存する方法

コード
```### 前提・実現したいこと
xy平面上にマウスでクリックすることでその座標点にプロットして座標点の値を保存し,最終的にはその座標点を使って計算を行いたいと考えています.

プロットを行うことは可能となり,座標点の値を表示することはできるのですが値の保存ができないです.
次のコードをどのように変えれば値の保存が可能となり以降の計算で使用できるようになるのでしょうか.

### 該当のソースコード
import matplotlib.pyplot as plt
import numpy as np

def main():
    
    fig = plt.figure()
    ax = fig.add_subplot(111)
    ClickToDrawPoints(ax).show()
    

    
    

class ClickToDrawPoints(object):
    def __init__(self, ax):
        self.ax = ax
        self.fig = ax.figure
        self.xy = []
        self.points = ax.scatter([], [], s=200, color='red', picker=20)
        self.fig.canvas.mpl_connect('button_press_event', self.on_click)
        
        
        
        
        

     def on_click(self, event):
        if event.inaxes is None:
            return
        
        self.xy.append([event.xdata, event.ydata])
        self.points.set_offsets(self.xy)
        self.ax.draw_artist(self.points)
        self.fig.canvas.blit(self.ax.bbox)
  
    def show(self):
        self.ax.set_xticks(np.linspace(0, 4, 5))
        self.ax.set_yticks(np.linspace(0, 4, 5))
        self.ax.set_aspect('equal')
        plt.show()

main()

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

回答2件

ベストアンサー

座標自体は、self.xy に格納されておりますので、グラフ上にButtonを配置してButtonをClickしたタイミングでこのデータをファイルを書き出すというのはどうでしょうか

Python
1import matplotlib.pyplot as plt
2from matplotlib.widgets import Button
3import numpy as np
4import csv
5
6
7def main():
8    fig = plt.figure()
9    ClickToDrawPoints(fig).show()
10
11class ClickToDrawPoints(object):
12    def __init__(self, fig):
13        self.fig = fig
14        self.ax1 = fig.add_axes([0.1,0.1,0.8,0.8])
15        self.ax2 = fig.add_axes([0.45,0.925,0.1,0.05])
16        self.xy = []
17        self.points = self.ax1.scatter([], [], s=200, color='red', picker=20)
18        self.ax1.figure.canvas.mpl_connect('button_press_event', self.on_click)
19        self.bt = Button(self.ax2, "SAVE")
20        self.bt.on_clicked(self.on_button_click)
21
22    def on_click(self, event):
23        if event.inaxes is not self.ax1:
24            return
25        self.xy.append([event.xdata, event.ydata])
26        self.points.set_offsets(self.xy)
27        self.ax1.draw_artist(self.points)
28        self.ax1.figure.canvas.blit(self.ax1.bbox)
29
30    def on_button_click(self, event):
31        with open("data.csv", "w") as f:
32            writer = csv.writer(f, lineterminator="\n")
33            writer.writerows(self.xy)
34
35    def show(self):
36        self.ax1.set_xticks(np.linspace(0, 4, 5))
37        self.ax1.set_yticks(np.linspace(0, 4, 5))
38        self.ax1.set_aspect('equal')
39        plt.show()
40
41main()