根據名稱選擇頂點

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']