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 只返回顶点序列的第一个顶点。