根据名称选择顶点

Question

我使用以下代码创建图形并添加节点。

g = Graph() 
g.add_vertices(3) 
g.vs["name"] = ["ABC","ABCD","ACD"] 

我想从顶点名称包含'AB'的图中选择顶点。从上面的图表中，它应该返回“ABC”和“ABCD”。有没有办法做到这一点？

Answer 1

由于您尚未指定图形将包含的节点数量，因此我会假设您不需要缩放。您可以使用List Comprehension：

a = ["ABC","ABCD","ACD"] #List of vertices 
#or in your case a = g.vs['name'] 

b = ["AB"]    #Pattern to be matched 

match =[vert for patt in b for vert in a if patt in vert] 

print match 
#['ABC', 'ABCD']