matplotlib の描画に関して

Question

matplotlibでクリックをした座標にマーカーを打つプログラムですが、凡例に表示されているクリック回数とマーカーをひとつ前に戻るような機能をもたせることは可能でしょうか。（例えば100回プロットしなくてはいけない業務があったとして、ひとつでも相違の位置にプロットしてしまうと始めからやり直す手間がある）

ご存じの方、参考になるようなサイトを知っている方はご教授願います。よろしくお願いします。

import matplotlib.pyplot as plt 
plt.rcParams['font.family'] = 'MS Gothic'
markers = list('ov^<>12348sp*hH+xD')
colors = list('bgrcmyk')
def motion(event):
    global i, j, cnt
    x, y = event.xdata, event.ydata
    plt.plot(
        [x], [y], color=colors[j], marker=markers[i], markersize=12)
    i, j = (i + 1) % len(markers), (j + 1) % len(colors)
    cnt += 1
    leg = plt.legend([f'クリック回数 {cnt}'], markerscale=0, handletextpad=-2.0, loc='upper right')
    leg.legendHandles[0].set_visible(False)
    plt.draw()

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

Comments

[1T2R3M4]

コピペで済ましているからわからないのでは。

[AN3000]

わからない原因は私の経験、スキル不足のためです。
私が書いているコードはこの質問には不要な部分が多いため、以前の回答者様のコードをそのまま載せています。

Answer 1

参考になるようなサイト

Event handling and picking - Matplotlib 3.1.0 documentation が参考になるかと思います。ここに記載されているサンプルプログラムを元に全体を書き直してみたものが以下になります(右クリックで最後にプロットした図形オブジェクトを削除)。

import matplotlib.pyplot as plt 
from itertools import cycle

plt.rcParams['font.family'] = 'MS Gothic'

class PlaceObjects:
    def __init__(self, fig):
        self.markers = cycle('ov^<>12348sp*hH+xD')
        self.colors = cycle('bgrcmyk')
        self.fig = fig
        self.ax = self.fig.axes[0]
        self.fig.canvas.mpl_connect('button_press_event', self.on_press)

    def on_press(self, event):
        x, y = event.xdata, event.ydata
        if event.button == 1:   # left click
            self.add_object(x, y)
        elif event.button == 3: # right click
            self.remove_object()
        # update legend
        n = len(self.ax.lines)
        leg = self.ax.legend(
            [f'オブジェクト数 {n}'], handletextpad=-2.0, loc='upper right')
        if n: leg.legendHandles[0].set_visible(False)
        self.fig.canvas.draw()

    def add_object(self, x, y):
        self.ax.plot([x], [y], markersize=12,
            color=next(self.colors), marker=next(self.markers))

    def remove_object(self):
        if self.ax.lines:
            self.ax.lines[-1].remove()

if __name__ == '__main__':
    fig, ax = plt.subplots()
    ax.set_xlim(0, 1)
    ax.set_ylim(0, 1)

    _ = PlaceObjects(fig)

    plt.show()

Comments

[AN3000]

参考サイトまで提示してくださり、ありがとうございます！無事に自身のコードに組み込み、動作確認できました。グーグル翻訳などを使ってみていこうと思います！ありがとうございました！！

Answer 2

import matplotlib.pyplot as plt 
plt.rcParams['font.family'] = 'MS Gothic'
markers = list('ov^<>12348sp*hH+xD')
colors = list('bgrcmyk')
ln = [] # artist記憶用

def motion(event):
    global i, j, cnt
    if event.button == 3: # 右クリック->1つ前を削除
        if len(ln) > 0:
            ln[-1].pop(0).remove()
            ln.pop()
            cnt -= 1
            i, j = (i - 1) % len(markers), (j - 1) % len(colors)
            if len(ln) == 0: # 座標数が0になったら凡例非表示
                ax.get_legend().remove()
            else:
                leg = ax.legend([f'クリック回数 {cnt}'], markerscale=0, handletextpad=-2.0, loc='upper right')
                leg.legendHandles[0].set_visible(False)
            fig.canvas.draw()
    else: # それ以外->登録
        x, y = event.xdata, event.ydata
        ln.append(ax.plot([x], [y], color=colors[j], marker=markers[i], markersize=12))
        i, j = (i + 1) % len(markers), (j + 1) % len(colors)
        cnt += 1
        leg = ax.legend([f'クリック回数 {cnt}'], markerscale=0, handletextpad=-2.0, loc='upper right')
        leg.legendHandles[0].set_visible(False)
        fig.canvas.draw()

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

こんな感じでしょうか？
button_press_event のうち、event.button == 3（＝右クリック）の時のみ最新の座標点を削除するようにしてあります。
また、補足になりますが、plt.subplots()で figure と axes を作成していながら plt 操作でしたので、figure と axes での操作をメインにしておきました。

Comments

[AN3000]

補足と解説までしてくださりありがとうございます。ベストアンサーは別の方を選ばせていただきましたが、すごくわかりやすく感じました。ほんとうにありがとうございました。

Answer 3

KB 操作で戻すでよろしければ

import matplotlib.pyplot as plt 
plt.rcParams['font.family'] = 'MS Gothic'
markers = list('ov^<>12348sp*hH+xD')
colors = list('bgrcmyk')
dots = []
def motion(event):
    global i, j, cnt
    x, y = event.xdata, event.ydata
    dot, = plt.plot([x], [y], color=colors[j], marker=markers[i], markersize=12)
    dots.append(dot)
    i, j = (i + 1) % len(markers), (j + 1) % len(colors)
    cnt += 1
    leg = plt.legend([f'クリック回数 {cnt}'], markerscale=0, handletextpad=-2.0, loc='upper right')
    leg.legendHandles[0].set_visible(False)
    plt.draw()

def key_press(event):
    global cnt
    if dots and cnt > 1:
        dots.pop().remove()
        cnt -= 1
        leg = plt.legend([f'クリック回数 {cnt}'], markerscale=0, handletextpad=-2.0, loc='upper right')
        leg.legendHandles[0].set_visible(False)
        plt.draw()

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

右クリック版追加しておきます。

import matplotlib.pyplot as plt 
plt.rcParams['font.family'] = 'MS Gothic'
markers = list('ov^<>12348sp*hH+xD')
colors = list('bgrcmyk')
dots = []
def motion(event):
    global i, j, cnt
    if event.button == 3:  # 右クリック
        if dots and cnt > 0:
            dots.pop().remove()
            cnt -= 1
    else:
        x, y = event.xdata, event.ydata
        dot, = plt.plot([x], [y], color=colors[j], marker=markers[i], markersize=12)
        dots.append(dot)
        i, j = (i + 1) % len(markers), (j + 1) % len(colors)
        cnt += 1
    leg = plt.legend([f'クリック回数 {cnt}'], markerscale=0, handletextpad=-2.0, loc='upper right')
    if leg.legendHandles:
        leg.legendHandles[0].set_visible(False)
    plt.draw()

plt.xlim(0, 1)
plt.ylim(0, 1)
i = j = cnt = 0
plt.connect('button_press_event', motion)
plt.show()

Comments

[AN3000]

kb操作とクリック操作両方教えていただきありがとうございます。無事、自身のコードに組み込むことができました！本当にありがとうございます！！