本文整理汇总了Python中apgl.graph.DictGraph.DictGraph.getVertex方法的典型用法代码示例。如果您正苦于以下问题：Python DictGraph.getVertex方法的具体用法？Python DictGraph.getVertex怎么用？Python DictGraph.getVertex使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类apgl.graph.DictGraph.DictGraph的用法示例。
在下文中一共展示了DictGraph.getVertex方法的4个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: testGetVertex
# 需要导入模块: from apgl.graph.DictGraph import DictGraph [as 别名]
# 或者: from apgl.graph.DictGraph.DictGraph import getVertex [as 别名]
def testGetVertex(self):
dictGraph = DictGraph(True)
dictGraph.addEdge(1, 2, 12)
dictGraph.addEdge(1, 3, 18)
dictGraph.setVertex(5, 12)
self.assertEquals(dictGraph.getVertex(1), None)
self.assertEquals(dictGraph.getVertex(2), None)
self.assertEquals(dictGraph.getVertex(3), None)
self.assertEquals(dictGraph.getVertex(5), 12)
self.assertRaises(ValueError, dictGraph.getVertex, 4)
#Directed graphs
dictGraph = DictGraph(False)
dictGraph.addEdge(1, 2, 12)
dictGraph.addEdge(1, 3, 18)
dictGraph.setVertex(5, 12)
self.assertEquals(dictGraph.getVertex(1), None)
self.assertEquals(dictGraph.getVertex(2), None)
self.assertEquals(dictGraph.getVertex(3), None)
self.assertEquals(dictGraph.getVertex(5), 12)
self.assertRaises(ValueError, dictGraph.getVertex, 4)示例2: testAddVertex
# 需要导入模块: from apgl.graph.DictGraph import DictGraph [as 别名]
# 或者: from apgl.graph.DictGraph.DictGraph import getVertex [as 别名]
def testAddVertex(self):
dictGraph = DictGraph(True)
dictGraph.addEdge(1, 2, 12)
dictGraph.addEdge(1, 3, 18)
dictGraph.setVertex(5, 12)
self.assertEquals(dictGraph.getVertex(5), 12)
dictGraph.setVertex(5, 22)
self.assertEquals(dictGraph.getVertex(5), 22)
dictGraph.addEdge(5, 11, 18)
self.assertEquals(dictGraph.getVertex(5), 22)示例3: testSubgraph
# 需要导入模块: from apgl.graph.DictGraph import DictGraph [as 别名]
# 或者: from apgl.graph.DictGraph.DictGraph import getVertex [as 别名]
def testSubgraph(self):
graph = DictGraph()
graph.addEdge(0, 1)
graph.addEdge(0, 2)
graph.addEdge(0, 3)
graph.addEdge(1, 2)
graph.addEdge(2, 3)
graph.setVertex(0, "abc")
graph.setVertex(3, "cde")
self.assertEquals(graph.getNumEdges(), 5)
subgraph = graph.subgraph([0, 1, 2])
self.assertEquals(subgraph.getNumVertices(), 3)
self.assertEquals(subgraph.getNumEdges(), 3)
self.assertEquals(subgraph.isUndirected(), True)
self.assertEquals(subgraph.getEdge(0, 1), 1)
self.assertEquals(subgraph.getEdge(0, 2), 1)
self.assertEquals(subgraph.getEdge(1, 2), 1)
self.assertEquals(subgraph.getVertex(0), "abc")
#Check the original graph is fine
self.assertEquals(graph.getNumVertices(), 4)
self.assertEquals(graph.getNumEdges(), 5)
self.assertEquals(graph.getVertex(0), "abc")
self.assertEquals(graph.getVertex(3), "cde")
#Now a quick test for directed graphs
graph = DictGraph(False)
graph.addEdge(0, 1)
graph.addEdge(0, 2)
graph.addEdge(0, 3)
graph.addEdge(1, 2)
graph.addEdge(2, 3)
subgraph = graph.subgraph([0, 1, 2])
self.assertEquals(subgraph.getNumEdges(), 3)
self.assertEquals(subgraph.isUndirected(), False)
self.assertEquals(subgraph.getEdge(0, 1), 1)
self.assertEquals(subgraph.getEdge(0, 2), 1)
self.assertEquals(subgraph.getEdge(1, 2), 1)示例4: __init__
# 需要导入模块: from apgl.graph.DictGraph import DictGraph [as 别名]
# 或者: from apgl.graph.DictGraph.DictGraph import getVertex [as 别名]
def __init__(self, minGraphSize=500, maxGraphSize=None, dayStep=30):
dataDir = PathDefaults.getDataDir() + "cluster/"
edgesFilename = dataDir + "Cit-HepTh.txt"
dateFilename = dataDir + "Cit-HepTh-dates.txt"
#Note the IDs are integers but can start with zero so we prefix "1" to each ID
edges = []
file = open(edgesFilename, 'r')
file.readline()
file.readline()
file.readline()
file.readline()
for line in file:
(vertex1, sep, vertex2) = line.partition("\t")
vertex1 = vertex1.strip()
vertex2 = vertex2.strip()
edges.append([vertex1, vertex2])
#if vertex1 == vertex2:
# print(vertex1)
file.close()
logging.info("Loaded edge file " + str(edgesFilename) + " with " + str(len(edges)) + " edges")
#Keep an edge graph
graph = DictGraph(False)
graph.addEdges(edges)
logging.info("Created directed citation graph with " + str(graph.getNumEdges()) + " edges and " + str(graph.getNumVertices()) + " vertices")
#Read in the dates articles appear in a dict which used the year and month
#as the key and the value is a list of vertex ids. For each month we include
#all papers uploaded that month and those directed cited by those uploads.
startDate = datetime.date(1990, 1, 1)
file = open(dateFilename, 'r')
file.readline()
numLines = 0
subgraphIds = []
for line in file:
(id, sep, date) = line.partition("\t")
id = id.strip()
date = date.strip()
inputDate = datetime.datetime.strptime(date.strip(), "%Y-%m-%d")
inputDate = inputDate.date()
if graph.vertexExists(id):
tDelta = inputDate - startDate
graph.vertices[id] = tDelta.days
subgraphIds.append(id)
#If a paper cites another, it must have been written before
#the citing paper - enforce this rule.
for neighbour in graph.neighbours(id):
if graph.getVertex(neighbour) == None:
graph.setVertex(neighbour, tDelta.days)
subgraphIds.append(neighbour)
elif tDelta.days < graph.getVertex(neighbour):
graph.setVertex(neighbour, tDelta.days)
numLines += 1
file.close()
subgraphIds = set(subgraphIds)
graph = graph.subgraph(list(subgraphIds))
logging.debug(graph)
logging.info("Loaded date file " + str(dateFilename) + " with " + str(len(subgraphIds)) + " dates and " + str(numLines) + " lines")
W = graph.getSparseWeightMatrix()
W = W + W.T
vList = VertexList(W.shape[0], 1)
vList.setVertices(numpy.array([graph.getVertices(graph.getAllVertexIds())]).T)
#Note: we have 16 self edges and some two-way citations so this graph has fewer edges than the directed one
self.graph = SparseGraph(vList, W=W)
logging.debug(self.graph)
#Now pick the max component
components = self.graph.findConnectedComponents()
self.graph = self.graph.subgraph(components[0])
logging.debug("Largest component graph: " + str(self.graph))
self.minGraphSize = minGraphSize
self.maxGraphSize = maxGraphSize
self.dayStep = dayStep