Python 从数据帧生成列表

我有一个像熊猫一样的数据框

   node    centroid
    1         1
    2         2
    3         4
    5         6
    7         2
    4         1
    8         4
    10        1

如何通过此数据集创建列表生成器，使具有相同质心的节点位于相同的列表中？

---

我需要像find clique return（）这样的结果，您可以使用

yield

来生成生成器

import pandas as pd
import random
df = pd.DataFrame({"node": [random.randint(1, 11) for _ in range(8)],
                   "centroid": [random.randint(1, 5) for _ in range(8)]})

def list_gen(df):
  for x in df.centroid.unique():
    yield df[df["centroid"] == x].node.values

[x for x in list_gen(df)]

这将提供numpy数组，但您只需将它们传递到

list

即可将它们解构回常规的

list