如何在matplot图中显示数据

Question

我正在尝试在jupyter notebook中创建一个交互式图，但我不确切知道如何实现它。拥有一个数据帧我运行一个简单的回归，然后绘制，以查看分布。我希望能够悬停其中一个点并获得与此点相关的数据。我怎样才能做到这一点？现在我只能制作静态情节

import pandas as pd
from sklearn import linear_model
%matplotlib inline
import matplotlib
import matplotlib.pyplot as plt

net = pd.read_csv("network_ver_64.csv")
net = net[net.AWDT12 > 0]

x = net.LOAD_DAILY.values
y = net.AWDT12.values
x_lenght = int(x.shape[0])
y_lenght = int(y.shape[0])
x = x.reshape(x_lenght, 1)
y = y.reshape(y_lenght,1)
regr = linear_model.LinearRegression()
regr.fit(x, y)

plt.scatter(x, y,  color='black')
plt.plot(x, regr.predict(x), color='blue', linewidth=1)
plt.xticks(())
plt.yticks(())
plt.show()

Answer 1

首先，很明显%matplotlib inline后端不允许交互，因为它是内联的（在图中是图像的意义上）。

然而，即使在笔记本中，您也可以使用%matplotlib notebook后端进行交互。已经实现了基本的悬停：在画布中移动鼠标会在右下角的数据坐标中显示当前鼠标位置。

当然，您可以通过编写一些自定义代码来获得更复杂的功能。例如。我们可以稍微修改picking example如下：

import matplotlib.pyplot as plt
%matplotlib notebook
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(111)
ax.set_title('click on points')

line, = ax.plot(np.random.rand(100), 'o', picker=5)  # 5 points tolerance
text = ax.text(0,0,"")
def onpick(event):
    thisline = event.artist
    xdata = thisline.get_xdata()
    ydata = thisline.get_ydata()
    ind = event.ind
    text.set_position((xdata[ind], ydata[ind]))
    text.set_text(zip(xdata[ind], ydata[ind]))

fig.canvas.mpl_connect('pick_event', onpick)

plt.show()

现在显示鼠标点击的点的数据坐标。

您可以随意使用标准的matplotlib工具，使其适应您喜欢的任何情况。