快捷導航

matplotlib交互式數(shù)據(jù)光標實現(xiàn)(mplcursors)

更新時間：2021年01月13日 09:31:03 作者：mighty13

這篇文章主要介紹了matplotlib交互式數(shù)據(jù)光標實現(xiàn)(mplcursors),文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

簡介

mplcursors包也可以為matplotlib提供交互式的數(shù)據(jù)光標（彈出式注釋框），它的靈感來源于mpldatacursor包，可以認為是基于mpldatacursor包的二次開發(fā)。
相對于mpldatacursor包，mplcursors包最大的特點就是提供了一些相對底層的API，這樣功能實現(xiàn)更加靈活。

安裝

pip install mplcursors

基本應(yīng)用

mplcursors包的基本應(yīng)用方法與mpldatacursor包類似，直接應(yīng)用cursor函數(shù)即可。

基本操作方法

鼠標左鍵單擊圖表數(shù)據(jù)元素時會彈出文本框顯示最近的數(shù)據(jù)元素的坐標值。
鼠標右鍵單擊文本框取消顯示數(shù)據(jù)光標。
按d鍵時切換顯示\關(guān)閉數(shù)據(jù)光標。

案例源碼

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

data = np.outer(range(10), range(1, 5))

fig, ax = plt.subplots()
lines = ax.plot(data)
ax.set_title("Click somewhere on a line.\nRight-click to deselect.\n"
       "Annotations can be dragged.")

mplcursors.cursor(lines) # or just mplcursors.cursor()

plt.show()

mplcursors自定義應(yīng)用

mpldatacursor包中自定義功能主要通過向datacursor函數(shù)傳遞實參實現(xiàn)。
mplcursors包中的cursor函數(shù)對標mpldatacursor包中的datacursor函數(shù)，但是在參數(shù)上發(fā)生了變化，保留了artists、hover、bindings、multiple、highlight等類似參數(shù)。
mplcursors包增加Selection對象（底層為namedtuple）表示選擇的數(shù)據(jù)元素的屬性。
當選中某個數(shù)據(jù)點時，可以通過添加（add）或刪除（remove）事件觸發(fā)、注冊回調(diào)函數(shù)實現(xiàn)功能，回調(diào)函數(shù)只有一個參數(shù)，及選擇的數(shù)據(jù)點。
在注冊回調(diào)函數(shù)時，mplcursors包支持使用裝飾器。

mpldatacursor與mplcursors API對比

下面以修改顯示文本信息為例對比下mpldatacursor與mplcursors的不同實現(xiàn)方式。

在這里插入圖片描述

mpldatacursor實現(xiàn)方式

import matplotlib.pyplot as plt
import numpy as np
from mpldatacursor import datacursor

ax=plt.gca()
labels = ["a", "b", "c"]
for i in range(3):
  ax.plot(i, i,'o', label=labels[i])

datacursor(formatter='{label}'.format)
plt.show()

mplcursors實現(xiàn)方式一

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

ax=plt.gca()
lines = ax.plot(range(3), range(3), "o")
labels = ["a", "b", "c"]
cursor = mplcursors.cursor(lines)
cursor.connect(
  "add", lambda sel: sel.annotation.set_text(labels[sel.target.index]))

plt.show()

mplcursors實現(xiàn)方式二

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

ax=plt.gca()
lines = ax.plot(range(3), range(3), "o")
labels = ["a", "b", "c"]
cursor = mplcursors.cursor(lines)

@cursor.connect("add")
def on_add(sel):
  sel.annotation.set_text(labels[sel.target.index])
plt.show()