pyqtgraph获取节点的文本并在MouseClick上更改颜色
问题描述:
我发现this示例来自pyqtgraph github回购。pyqtgraph获取节点的文本并在MouseClick上更改颜色
我想与pyqtgraph
具有以下功能创建一个交互式(网络)图:
- 当用户点击一个节点上,他/她将获得的 的“text',信息是节点。
- 当一个节点被点击节点的颜色,以及它的所有邻居(边缘和节点)的 将改变(例如黄色)
- 允许多个选择
代码:
# -*- coding: utf-8 -*-
"""
Simple example of subclassing GraphItem.
"""
import initExample ## Add path to library (just for examples; you do not need this)
import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui
import numpy as np
# Enable antialiasing for prettier plots
pg.setConfigOptions(antialias=True)
w = pg.GraphicsWindow()
w.setWindowTitle('pyqtgraph example: CustomGraphItem')
v = w.addViewBox()
v.setAspectLocked()
class Graph(pg.GraphItem):
def __init__(self):
self.dragPoint = None
self.dragOffset = None
self.textItems = []
pg.GraphItem.__init__(self)
self.scatter.sigClicked.connect(self.clicked)
def setData(self, **kwds):
self.text = kwds.pop('text', [])
self.data = kwds
if 'pos' in self.data:
npts = self.data['pos'].shape[0]
self.data['data'] = np.empty(npts, dtype=[('index', int)])
self.data['data']['index'] = np.arange(npts)
self.setTexts(self.text)
self.updateGraph()
def setTexts(self, text):
for i in self.textItems:
i.scene().removeItem(i)
self.textItems = []
for t in text:
item = pg.TextItem(t)
self.textItems.append(item)
item.setParentItem(self)
def updateGraph(self):
pg.GraphItem.setData(self, **self.data)
for i,item in enumerate(self.textItems):
item.setPos(*self.data['pos'][i])
def mouseDragEvent(self, ev):
if ev.button() != QtCore.Qt.LeftButton:
ev.ignore()
return
if ev.isStart():
# We are already one step into the drag.
# Find the point(s) at the mouse cursor when the button was first
# pressed:
pos = ev.buttonDownPos()
pts = self.scatter.pointsAt(pos)
if len(pts) == 0:
ev.ignore()
return
self.dragPoint = pts[0]
ind = pts[0].data()[0]
self.dragOffset = self.data['pos'][ind] - pos
elif ev.isFinish():
self.dragPoint = None
return
else:
if self.dragPoint is None:
ev.ignore()
return
ind = self.dragPoint.data()[0]
self.data['pos'][ind] = ev.pos() + self.dragOffset
self.updateGraph()
ev.accept()
def clicked(self, pts):
print("clicked: %s" % pts)
g = Graph()
v.addItem(g)
## Define positions of nodes
pos = np.array([
[0,0],
[10,0],
[0,10],
[10,10],
[5,5],
[15,5]
], dtype=float)
## Define the set of connections in the graph
adj = np.array([
[0,1],
[1,3],
[3,2],
[2,0],
[1,5],
[3,5],
])
## Define the symbol to use for each node (this is optional)
symbols = ['o','o','o','o','t','+']
## Define the line style for each connection (this is optional)
lines = np.array([
(255,0,0,255,1),
(255,0,255,255,2),
(255,0,255,255,3),
(255,255,0,255,2),
(255,0,0,255,1),
(255,255,255,255,4),
], dtype=[('red',np.ubyte),('green',np.ubyte),('blue',np.ubyte),('alpha',np.ubyte),('width',float)])
## Define text to show next to each symbol
texts = ["Point %d" % i for i in range(6)]
## Update the graph
g.setData(pos=pos, adj=adj, pen=lines, size=1, symbol=symbols, pxMode=False, text=texts)
## Start Qt event loop unless running in interactive mode or using pyside.
if __name__ == '__main__':
import sys
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
QtGui.QApplication.instance().exec_()
上述代码返回以下信息,被点击的节点时:
<pyqtgraph.graphicsItems.ScatterPlotItem.ScatterPlotItem object at 0x7f1e51657558>
这实际上是整个图形项目(因为点击信号在整个图形上被调用)。
现在:我怎样才能在每个节点上调用MouseClicking功能,获取该节点的文本并在点击事件中更改它的(及其邻居)颜色?
Another example显示在另一个绘图中更改点击曲线的颜色。这里,点击信号在每条曲线上被调用。我试图用这个作为一个起点,在下面第一个提到的代码示例中实现类似的东西,但老实说,我甚至不知道如何到达pyqtgraph图形对象中的单个节点([我可能在这里是错的],节点只能通过它们的位置来定义)。
一如既往,在这里的任何帮助将非常感激。
编辑:感谢kesumu的回答,我是能够得到一个点击的节点像这样的文本内容:
def clicked(self, scatter, pts):
data_list = scatter.data.tolist()
mypoint = [tup for tup in data_list if pts[0] in tup][0]
mypoint_index = data_list.index(mypoint)
mypoint_text = self.text[mypoint_index]
编辑II:
的一个更复杂的例子同样的问题可以发现here。
答
问题是您的clicked
函数是错误的。
它应该是这样的:
def clicked(self, scatter, pts):
print(scatter)
print(pts[0])
print("clicked: %s" % pts)
因为sigClicked
有两个参数:自我,点。这是它的定义:
sigClicked = QtCore.Signal(object, object) ## self, points
固定后,就可以获取单击点是这样的:
希望这对您有所帮助。
kesumu,非常感谢您的帮助,这确实非常有用!我现在可以获取点击节点以及它们的文本(将编辑问题如何获取文本)。但现在我坚持如何改变点击节点的颜色以及它的邻居......我会把这个问题放在另一个问题上,因为这个问题的主要关注点实际上是获取点击节点及其文本内容。再次谢谢你! – dliv