我使用以下代码创建一个图表并添加节点。
g = Graph()
g.add_vertices(3)
g.vs["name"] = ["ABC","ABCD","ACD"]
我想从图中选择顶点名称包含“AB”的顶点。从上图中,它应该返回“ABC”和“ABCD”。有办法做到这一点吗?
最佳答案
由于您没有指定图表将包含的节点数,因此我假设您不需要缩放。您可以使用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']
关于python - 根据名称选择顶点,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45429308/