scala - Scala Flatmap on Array 里面的值部分

Question

我有像 (a,b,Array(p1,p2,p3),c) 这样的 RDD，我想在值内的 Array 上应用 flatmap 以获得以下输出。(a,b,p1,c) (a,b,p2,c) (a,b,p3,c)

Answer 1

代码：

val arr = Array(("a1", "b1", Array("1", "2", "3")),("a1", "b1", Array("1", "2", "3")))
val rdd: RDD[(String, String, Array[String])] = sc.parallelize(arr)
val result = rdd.flatMap {
  case (first, second, third) => {
    third.map(x => (first, second, x))
}}

println("Results: " + result.collect.mkString("|"))

结果：

Results: (a1,b1,1)|(a1,b1,2)|(a1,b1,3)|(a1,b1,1)|(a1,b1,2)|(a1,b1,3)