加入 RDD 后删除括号

Question

我正在加入大量的 rdd，我想知道是否有一种通用的方法可以删除在每次加入时创建的括号。

这是一个小样本：

val rdd1 =  sc.parallelize(Array((1,2),(2,4),(3,6)))
val rdd2 =  sc.parallelize(Array((1,7),(2,8),(3,6)))
val rdd3 =  sc.parallelize(Array((1,2),(2,4),(3,6)))

val result = rdd1.join(rdd2).join(rdd3)

res: result: org.apache.spark.rdd.RDD[(Int, ((Int, Int), Int))] = Array((1,((2,7),2)), (3,((4,8),4)), (3,((4,8),6)), (3,((4,6),4)), (3,((4,6),6)))

我知道我可以使用地图

result.map((x) => (x._1,(x._2._1._1,x._2._1._2,x._2._2))).collect

Array[(Int, (Int, Int, Int))] = Array((1,(2,7,2)), (2,(4,8,4)), (3,(6,6,6)))

但是对于大量的 rdd，每个都包含许多元素，很快就很难使用这种方法

Answer 1

对于大量的 rdd，每个都包含许多元素，这种方法根本行不通，因为最大的内置元组仍然是 Tuple22。如果加入同质 RDD 某种类型的序列：

def joinAndMerge(rdd1: RDD[(Int, Seq[Int])], rdd2: RDD[(Int, Seq[Int])]) = 
  rdd1.join(rdd2).mapValues{ case (x, y) => x ++ y }

Seq(rdd1, rdd2, rdd3).map(_.mapValues(Seq(_))).reduce(joinAndMerge)

如果你只有三个 RDD，使用起来会更干净 cogroup:

rdd1.cogroup(rdd2, rdd3)
  .flatMapValues { case (xs, ys, zs) => for {
    x <- xs; y <- ys; z <- zs 
  } yield (x, y, z) }

如果值是异质的，则使用 DataFrames:

更有意义

def joinByKey(df1: DataFrame, df2: DataFrame) = df1.join(df2, Seq("k"))

Seq(rdd1, rdd2, rdd3).map(_.toDF("k", "v")).reduce(joinByKey)

加入 RDD 后删除括号

Removing parenthesis after joining RDDs

scala

apache-spark

rdd