Question

我是Scala和GraphX的新手，所以这可能是一个微不足道的问题。我试图反转图形每条边的方向，然后得到新反转图的每个顶点的邻接列表。输入格式为“FromNodeId \ t ToNodeId”

我设法使用以下代码反转边缘的方向：

object Problem2{
    def main(args: Array[String]){
    val inputFile:String = args(0)
    val outputFolder = args(1)
    val conf = new SparkConf().setAppName("Problem2").setMaster("local")
    val sc = new SparkContext(conf)
    val input = sc.textFile(inputFile)
    val graph = GraphLoader.edgeListFile(sc,inputFile)
    val newGraph = graph.reverse.edges
}
}

输出格式为

Edge(1,0,1)
Edge(3,1,1)
Edge(2,1,1)
Edge(2,0,1)

我的问题是， 1.问题是否有更有效的方法 2.如何从这里开始为每个顶点构建邻接列表？

Answer 1

这两个问题的答案是collectNeighbors and collectNeighborIds。

对于每个顶点，它计算邻居列表。注意EdgeDirection参数。

val vertices: RDD[(VertexId, String)] =  
    sc.parallelize(Array((1L,""), (2L,""), (4L,""), (6L,"")))


val edges: RDD[Edge[String]] = 
    sc.parallelize(Array(
        Edge(1L, 2L, ""),
        Edge(1L, 4L, ""),
        Edge(1L, 6L, "")))
val inputGraph = Graph(vertices, edges)

val verticesWithSuccessors: VertexRDD[Array[VertexId]] = 
    inputGraph.ops.collectNeighborIds(EdgeDirection.Out)

因此，当图形不可变时，您必须构建一个新图形。相同的边，新的顶点

val successorGraph = Graph(verticesWithSuccessors, edges)

这是一个部分结果：

val res = successorGraph.vertices.collect()
res: Array[(org.apache.spark.graphx.VertexId,
         Array[org.apache.spark.graphx.VertexId])] = 
  Array((4,Array()), (1,Array(2, 4, 6)), (6,Array()), (2,Array()))

Scala从源节点和目标节点的列表中获取每个顶点的邻接列表

1 个答案: