matplotlibのプロット

Question

matplotlibでマウスでクリックした座標にプロットするプログラムを組みたいのですが、情報が見つからず困っています。

具体的な内容としては
グラフ、もしくはmatplotlibで読み込んだ画像データにクリックしてプロットをしていくといったプログラムで、プロットした順番によって、プロットの形状や色を変えられる機能も付与したいと考えています。（最初のプロットは〇の青色、二番目は×の赤色など）

以下のコードではプロットはできるのですが、ひとつしかプロットできません。一部でもよいので教えていただけると幸いです。

import matplotlib.pyplot as plt 

def motion(event):  
    x = event.xdata
    y = event.ydata
    ln.set_data(x,y)
    # プロットを更新
    plt.draw()

plt.figure()
ln, = plt.plot([],[],'o')

plt.connect('button_press_event', motion)
plt.show()

Answer 1

データを append しましょう。

import matplotlib.pyplot as plt 
import numpy as np

def motion(event):  
    x = event.xdata
    y = event.ydata
    # append new data
    ln.set_xdata(np.append(ln.get_xdata(), x))
    ln.set_ydata(np.append(ln.get_ydata(), y))
    # プロットを更新
    plt.draw()

plt.figure()
ln, = plt.plot([],[],'o')

plt.connect('button_press_event', motion)
plt.show()

Comments

[AN3000]

ありがとうございます！！　うまくいきました！

Answer 2

点をずっと残すということなら、以下です。

import matplotlib.pyplot as plt 

markers = list('ov^<>12348sp*hH+xD')
colors = list('bgrcmyk')
def motion(event):  
    global i, j
    x = event.xdata
    y = event.ydata
    plt.plot([event.xdata],[event.ydata], marker=markers[i], color=colors[j], markersize=12)
    i = (i + 1) % len(markers)
    j = (j + 1) % len(colors)
    plt.draw()

fig, ax = plt.subplots()
ax.set_xlim(0, 1)
ax.set_ylim(0, 1)
i = j = 0
plt.connect('button_press_event', motion)
plt.show()

Comments

[AN3000]

ありがとうございました！！無事前に進めそうです！