Как сгладить коллекцию с помощью Spark/Scala?
В Scala я могу сгладить коллекцию, используя:
val array = Array(List("1,2,3").iterator,List("1,4,5").iterator)
//> array : Array[Iterator[String]] = Array(non-empty iterator, non-empty itera
//| tor)
array.toList.flatten //> res0: List[String] = List(1,2,3, 1,4,5)
Но как я могу выполнить подобное в Spark?
Чтение API-документа http://spark.apache.org/docs/0.7.3/api/core/index.html#spark.RDD, похоже, не является методом, который предоставляет эту функциональность?
Ответы
Ответ 1
Попробуйте flatMap с функцией идентификации (y => y
):
scala> val x = sc.parallelize(List(List("a"), List("b"), List("c", "d")))
x: org.apache.spark.rdd.RDD[List[String]] = ParallelCollectionRDD[1] at parallelize at <console>:12
scala> x.collect()
res0: Array[List[String]] = Array(List(a), List(b), List(c, d))
scala> x.flatMap(y => y)
res3: org.apache.spark.rdd.RDD[String] = FlatMappedRDD[3] at flatMap at <console>:15
scala> x.flatMap(y => y).collect()
res4: Array[String] = Array(a, b, c, d)
Ответ 2
Используйте flatMap
и identity
Predef
, это более читаемо, чем использование x => x
, например
myRdd.flatMap(identity)