如何在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后端进行交互.已经实现了基本的悬停:在画布中移动鼠标会在右下角的数据坐标中显示当前鼠标位置.

当然,您可以通过编写一些自定义代码来获得更复杂的功能.例如,我们可以稍微修改一下拣选示例,如下所示:

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工具使它更漂亮.