在散點(diǎn)圖上加入懸停註解

使用matplotlib 建立每個(gè)點(diǎn)代表一個(gè)特定物件的散佈圖時(shí)，它會(huì)很有幫助當(dāng)遊標(biāo)懸停在相應(yīng)點(diǎn)上時(shí)顯示物件的名稱。這允許用戶快速識(shí)別異常值或其他相關(guān)訊息，而不會(huì)用永久標(biāo)籤弄亂繪圖。

一種解決方案涉及使用註解功能建立一個(gè)標(biāo)籤，當(dāng)遊標(biāo)懸停在特定點(diǎn)上時(shí)該標(biāo)籤變得可見(jiàn)。以下是範(fàn)例程式碼片段：

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(1)

x = np.random.rand(15)
y = np.random.rand(15)
names = np.array(list("ABCDEFGHIJKLMNO"))
c = np.random.randint(1, 5, size=15)

norm = plt.Normalize(1, 4)
cmap = plt.cm.RdYlGn

fig, ax = plt.subplots()
sc = plt.scatter(x, y, c=c, s=100, cmap=cmap, norm=norm)

annot = ax.annotate("", xy=(0, 0), xytext=(20, 20), textcoords="offset points",
                    bbox=dict(boxstyle="round", fc="w"),
                    arrowprops=dict(arrowstyle="->"))
annot.set_visible(False)


def update_annot(ind):
    pos = sc.get_offsets()[ind["ind"][0]]
    annot.xy = pos
    text = "{}, {}".format(" ".join(list(map(str, ind["ind"]))),
                           " ".join([names[n] for n in ind["ind"]]))
    annot.set_text(text)
    annot.get_bbox_patch().set_facecolor(cmap(norm(c[ind["ind"][0]])))
    annot.get_bbox_patch().set_alpha(0.4)


def hover(event):
    vis = annot.get_visible()
    if event.inaxes == ax:
        cont, ind = sc.contains(event)
        if cont:
            update_annot(ind)
            annot.set_visible(True)
            fig.canvas.draw_idle()
        else:
            if vis:
                annot.set_visible(False)
                fig.canvas.draw_idle()

fig.canvas.mpl_connect("motion_notify_event", hover)

plt.show()

此程式碼定義了包含 15 個(gè)隨機(jī)點(diǎn)的散佈圖。每個(gè)點(diǎn)都與名稱數(shù)組中的一個(gè)名稱相關(guān)聯(lián)。 annotate 函數(shù)建立一個(gè)原本不可見(jiàn)的標(biāo)籤。

hover 函數(shù)定義為處理滑鼠移動(dòng)事件。當(dāng)遊標(biāo)懸停在某個(gè)點(diǎn)上時(shí)，它會(huì)檢查該點(diǎn)是否包含在散佈圖中。如果是這樣，它會(huì)使用物件的名稱和位置更新註釋，使其可見(jiàn)，並重新繪製圖形。當(dāng)遊標(biāo)離開(kāi)該點(diǎn)時(shí)，註釋被隱藏。

對(duì)於線圖而不是散點(diǎn)圖，可以採(cǎi)用相同的解決方案，如下所示：

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(1)

x = np.sort(np.random.rand(15))
y = np.sort(np.random.rand(15))
names = np.array(list("ABCDEFGHIJKLMNO"))

norm = plt.Normalize(1, 4)
cmap = plt.cm.RdYlGn

fig, ax = plt.subplots()
line, = plt.plot(x, y, marker="o")

annot = ax.annotate("", xy=(0, 0), xytext=(-20, 20), textcoords="offset points",
                    bbox=dict(boxstyle="round", fc="w"),
                    arrowprops=dict(arrowstyle="->"))
annot.set_visible(False)


def update_annot(ind):
    x, y = line.get_data()
    annot.xy = (x[ind["ind"][0]], y[ind["ind"][0]])
    text = "{}, {}".format(" ".join(list(map(str, ind["ind"]))),
                           " ".join([names[n] for n in ind["ind"]]))
    annot.set_text(text)
    annot.get_bbox_patch().set_alpha(0.4)


def hover(event):
    vis = annot.get_visible()
    if event.inaxes == ax:
        cont, ind = line.contains(event)
        if cont:
            update_annot(ind)
            annot.set_visible(True)
            fig.canvas.draw_idle()
        else:
            if vis:
                annot.set_visible(False)
                fig.canvas.draw_idle()

fig.canvas.mpl_connect("motion_notify_event", hover)

plt.show()

以上是如何在 Matplotlib 散點(diǎn)圖和線圖添加懸停註釋以在滑鼠懸停時(shí)顯示物件名稱？的詳細(xì)內(nèi)容。更多資訊請(qǐng)關(guān)注PHP中文網(wǎng)其他相關(guān)文章！