如何计算graphx中每条边的两个顶点的度数之和

How to compute the sum of degree of two vertexs in each edge in graphx

我有这样的图表:

val vertexArray = Array(
      (1L, ("Alice", 28)),
      (2L, ("Bob", 27)),
      (3L, ("Charlie", 65)),
      (4L, ("David", 42)),
      (5L, ("Ed", 55)))                               
val edges = sc.parallelize(Array(
                 Edge(1L, 2L, ""), 
                 Edge(1L, 3L, ""), 
                 Edge(2L, 4L, ""),
                 Edge(3L, 5L, ""),
                 Edge(2L, 3L, "")))
val graph = Graph(vertexArray, edges)

我想得到每条边上两个顶点的度数之和。 例如,节点 1L 有 2 个邻居,节点 2L 有 3 个邻居,那么我要得到的结果是“1L,2L,5”。整个结果是:

"1L, 2L, 5"
"1L, 3L, 5"
"2L, 4L, 4"
"3L, 5L, 4",
"2L, 3L, 6"

如何在 GraphX 中实现它?

首先你可以用它的总度数

替换每个顶点属性
val graphDegrees = graph.outerJoinVertices(graph.degrees)((_,_,deg) => deg.get)

最后,映射每个三元组并求和两个顶点的度数

val graphSum = graphDegrees.mapTriplets(t => t.srcAttr + t.dstAttr)

我们可以通过打印边来检查结果

graphSum.edges.collect.foreach(println)

给出

Edge(1,2,5)
Edge(1,3,5)
Edge(2,4,4)
Edge(2,3,6)
Edge(3,5,4)