Python - igraph 查找特定類別的所有頂點

Question

您好，我在 python 中的庫 igraph 存在以下問題。 我做了一個這樣的簡單樹：

from igraph import *
import numpy as np

p = Graph(directed=True)
p.degree(mode='in')

p.add_vertices(2)
p.vs["label"] = ["Anna", "Peter"]
p.vs["category"] = np.full(2,1)


p.add_vertices(1)

p.vs[-1]["label"] = "Michael"
p.vs[-1]["category"] = -1

result = p.vs.find(category=1)["label"]

print(result)

使用命令 p.vs.find(category=1)["label"] 我想找到所有類別與 1 相同的頂點。這意味着我需要一個列表 ["Anna", "Peter"]。 但由於某種原因，我的結果只是：

Anna

我怎樣才能解決這個問題？

Answer 1

自己找到答案，只需使用

p.vs.select(category=1)["label"]

find 只返回頂點序列的第一個頂點。