python matplotlibに関して

Question

matplotlibでクリックした場所にプロットをしていくスクリプトを作成しています。
そのなかで、クリックした回数をカウントするスクリプト（回数が表示されるのが好ましい）を組み込みたいと考えていますが、良い情報がなく困っています。

ちょっとした提案等でも構いませんので、回答していただけると嬉しいです。
ソースコードは以下です。ターミナルで動かしてみてください。

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()

Answer 1

凡例(legend)に表示する場合。

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}'], 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

[AN3000]

私が思い描いていた通りのものです！ありがとうございました！！

[lehshell]

マーカーがクリック文字に重なりそう。markerscale=0 が漏れていませんか？
leg = plt.legend([f'クリック回数 {cnt}'], markerscale=0, handletextpad=-2.0, loc='upper right')
なのでは？

[melian]

> lehshell
leg.legendHandles[0].set_visible(False) としていますのでマーカーは非表示になります。

[lehshell]

matplotlib 3.4.3 では
leg.legendHandles[0].set_visible(False)
があっても表示されていました。本来は消えないとおかしいのですね。

[melian]

なるほど、こちらは matplotlib 3.5.0 なのですがマーカーが消えますね。

[lehshell]

matplotlib 3.5.0 にして消えることを確認できました。Thanks!

Answer 2

タイトルに表示でよろしければ global 変数 cnt を用意して

    cnt += 1
    plt.title(f'クリック回数：{cnt:>3}', fontname='MS Gothic')
    plt.draw()

Comments

[AN3000]

ありがとうございます！思い描いていた通りのものができそうです！