pyspark中的图形框架bfs

pyspark中的图形框架bfs,pyspark,pyspark-dataframes,graphframes,Pyspark,Pyspark Dataframes,Graphframes,我不熟悉图形框架,在这里我尝试使用bfs查找员工层次结构。src和dst分别是empid和managerid `from graphframes import GraphFrame vertices = spark.createDataFrame([('1', 'Carter', 'Derrick', 50), ('2', 'May', 'Derrick', 26), ('

我不熟悉图形框架,在这里我尝试使用bfs查找员工层次结构。src和dst分别是empid和managerid

`from graphframes import GraphFrame
  vertices = spark.createDataFrame([('1', 'Carter', 'Derrick', 50), 
                              ('2', 'May', 'Derrick', 26),
                             ('3', 'Mills', 'Jeff', 80),
                              ('4', 'Hood', 'Robert', 65),
                              ('5', 'Banks', 'Mike', 93),
                             ('98', 'Berg', 'Tim', 28),
                             ('99', 'Page', 'Allan', 16)],
                             ['id', 'name', 'firstname', 'age'])
  edges = spark.createDataFrame([('2', '1'), 
                           ('3', '1'),
                          ('4', '1'),
                          ('5', '2'),
                           ('6', '3'),
                           ('7', '2'),
                           ('8', '5'),
                           ('9', '7')],
                          ['src', 'dst'])
 g1 = GraphFrame(vertices, edges)

  result5 = g1.bfs(fromExpr = "id == '5'", toExpr =  "id == '1'", maxPathLength = 10)
  result5.show()`
 **** Out put****
  +--------------------+------+--------------------+------+--------------------+
  |                from|    e0|                  v1|    e1|                  to|
  +--------------------+------+--------------------+------+--------------------+
  |[5, Banks, Mike, 93]|[5, 2]|[2, May, Derrick,...|[2, 1]|[1, Carter, Derri...|
  +--------------------+------+--------------------+------+--------------------+

到目前为止,我正在传递fromExpr=“id==”5”中的单个id,即如何从edges src列循环整个id列表,并将所有输出路径保存在结果数据框中。请告诉我您的意见。

这是否回答了您的问题?