2017-02-18 47 views
0

-------- -------- companytrades.csv试图实现networkx到graphicsscene

date,Company1,Company2 
1/2/2017,1001,1111 
1/3/2017,1001,1100 
1/4/2017,1111,1001 
1/5/2017,1100,1001 
1/6/2017,1011,1001 
1/7/2017,1001,1111 
1/8/2017,1111,1100 
1/9/2017,1100,1011 

,我发现了一组实施networkx类的这个很好的例子成graphicsscene GraphNetworkX.py

Node.py

Edge.py

我使用下面的代码来实现这三个模块

--------- -------- netxgui.py

import pandas as pd 
from numpy import * 
import sys 
from math import * 
from PyQt4 import QtCore,QtGui 
from PyQt4.QtCore import * 
from PyQt4.QtGui import * 
import networkx as nx 
import GraphNetworkX 
from x import * 

pf = pd.read_csv('/desktop/companytrades.csv',header=0,index_col=['date']) 

#setup the needed variables 
if __names__ == "__main__": 
    app = QtGui.QApplication(sys.argv) 
    app.setStyle("Plastique") 

class netx(QtGui.QDialog): 
    def __init__(self): 
     super(netx,self).__init__() 
     self.uiNX = Ui_Form() 
     self.uiNX.setupUi(self) 
     G = nx.from_pandas_dataframe(pf,'Company1','Company2',create_using=nx.DiGraph()) 
Pos = nx.spring_layout(g,scale=300) 
scene = GraphNetworkx.GraphGraphicsScene(G=g,pos=Pos) 
self.uiNX.neworkx_graphicsView.setScene(scene) 

NX = netx() 
NX.show() 
sys.exit(exit(app.exec_()) 
app.deleteLater() 

------- ------- x.py

from PyQt4 import QtCore,QtGui 
try: 
    _fromUtf8 = QtCore.QString.fromUtf8 
except AttributeError: 
    def _fromUtf8(s): 
     return s 

try: 
    _encoding = QtGui.QApplication.unicodeUTF8 
    def _translate(context, text, disambig): 
     return QtGui.QApplication.translate(context, text, disambig, _encoding) 
except AttributeError: 
    def _translate(context, text, disambig): 
     return QtGui.QApplication.translate(context, text, disambig) 

class Ui_Form(object): 
    def setupUi(self, Form): 
     Form.setObjectName(_fromUtf8("Form")) 
     Form.resize(1492,1029) 
     self.networkx_graphicsView = QtGui.QGraphicsView(Form) 
     self.networkx_graphicsView.setGeometry(QtCore.QRect(240,70,971,911)) 
     sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding) 
     sizePolicy.setHorizontalStretch(0) 
     sizePolicy.setVerticalStretch(5) 
     sizePolicy.setHeightForWidth(self.networkx_graphicsView.sizePolicy().hasHeightForWidth()) 
     self.networkx_graphicsView.setSizePolicy(sizePolicy) 
     self.networkx_graphicsView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff) 
     self.networkx_graphicsView.setObjectName(_fromUtf8("networkx_graphicsView")) 
     self.retranslateUi(Form) 
     QtCore.QMetaObject.connectSlotsByName(Form) 
    def retranslateUi(self,Form): 
     Form.setWindowTitle(_translate("Form","Form",None)) 

我在执行此操作时遇到问题。我遇到的问题是在GraphNetworkX下的drawGraph函数。看来打印功能与n正常工作,但我真的不明白用i替换它的理由。

i = 0 
    for n in G.nodes(): 
     #print str(pos[n][0]) + " " + str(pos[n][1])+ " "+str(n) 
     self.addNode(pos[i][0], pos[i][1], n) 
     i += 1 

我做了一个实验,用n代替i,并添加节点似乎工作,但后来我添加边的问题。

for e in G.edges(): 
     node1 = self.nodes[e[0]] 
     node2 = self.nodes[e[1]] 
     self.addEdge(node1, node2) 

它口口声声说E [0]出range..considering的Nodes.py功能是把self.nodes成Nodes.nodes对象的名单,我想这也许就是占位符的每个对象,但我不太确定。然后我试图使这一变化:

node1 = e[0] 
node2 = e[1] 

但后来我得到的错误在Edge.py指出

x1 = node1.centerX() 

的整数不具备的功能的centerX()

我有点不知所措。有人能帮助我理解这一点,并帮助我实现这个目标吗?

回答

1

更简单的问题可能有助于获得答案。 (例如,您的x.py似乎与最后的问题似乎没有关系)

为了让您开始,您可能会遭受某些命名空间重载(重命名相同的对象)。在netxgui.py你有import networkx as nx但后来几行你宣布一个类nx(QtGui.QDialog)。在此之后,您尝试拨打nx.spring_layout(),您大概想要从networkx而不是您自定义的NX类。您尝试访问的结构可能存在于networkx Graph实例中,但不在QDialog实例中?

一般来说,避免使用from examplemodule import *会让您在追踪错误或开发代码时更轻松。

--- 编辑 ---

我仔细看了看,问题是因为在networkx.Graph/.DiGraph等,并在GraphGraphicsScene使用的不同表示。 Networkx使用下面的字典,用于存储节点(以及边)。 GraphGraphicsScene使用节点的列表

快速和肮脏的解决方案:这些发生是兼容的如果您的节点标记从0开始,并有一个连续的序列。

稍微更稳健。将生成自己的图形这样的,它可以实现如下:

def re_index_digraph(G_orig): 
    remap = {i: n for (i,n) in enumerate(G_orig.nodes())} 
    inv_remap = {v:k for k,v in remap.items()} 
    G = nx.DiGraph() 
    G.add_nodes_from(remap) 
    remap_edges = [(inv_remap[src], inv_remap[sink]) for (src, sink) in G_orig.edges()] 
    G.add_edges_from(remap_edges) 
    return G 

G_orig = nx.from_pandas_dataframe(pf,'Company1','Company2',create_using=nx.DiGraph()) 
G = re_index_digraph(G_orig) 

再进行与G你面前了。 (如果除了简单的节点名称和边缘之外还有属性,则需要相应地扩展重映射函数)。

除此之外,您可能需要投入大量精力开发GraphGraphicsScene实现。

+0

感谢您的意见。我会做更正,并得到结果..我怀疑这将完全解决问题。我添加了x.py,因为我正在处理PyQt4,并且从经验中知道,处理GUI的环境和组件的行为会发生变化。 – Daniel

+0

仍然无法正常工作..请帮助。 – Daniel

+0

什么不行?我想你没有使用上面的确切代码(netxgui.py),因为它不能按原样运行。但是,如果你是,至少你需要做一些缩进的类定义 – Bonlenfum