我正在编写一个Spark代码,我需要将类型为(String,(String,String))
的RDD转换为((String,String),String
)。
我有以下输入文本文件:
Language,Language-code,TotalViewsInThatLang
English,en,10965376,"Main_Page",2938355
Russian,ru,1925718,"%D0%97%D0%B0%D0%B3%D0%BB,915495
Spanish,es,1010810,"Wikipedia:Portada",13603
我已经创建了一个RDD,如下所示:
val line = sc.textFile(inputFile)
val nrdd = line.map(x=>(x.split(",")(0),(x.split(",")(1),x.split(",")(2))))
nrdd: org.apache.spark.rdd.RDD[(String, (String, String))] = MapPartitionsRDD[2] at map at <console>:26
从这里我想使用case
函数创建类型为((String,String),String)
的RDD。
如何在map
中使用case
语句?
编辑
当我试图使用case函数时,我得到以下错误:
scala> val frdd = nrdd.map( {case(x,(y,z))=>((x,y),z))})
<console>:1: error: ';' expected but ')' found.
val frdd = nrdd.map({case(x,(y,z))=>((x,y),z))})
^
除非我误解了你的问题,否则你需要这样:
val list: List[((String, String), String)] = List((("a1", "b1"), "c1"), (("a2", "b2"), "c2"))
val res = list.map { case ((a, b), c) => (a, (b, c)) }
println(res) // List((a1,(b1,c1)), (a2,(b2,c2)))
由于您的RDD是Pair RDD,您可以使用Keyed RDD
中的swap
。
示例代码:
val keyRDD = sc.parallelize(List((("a1", "b1"), "c1"), (("a2", "b2"), "c2")), 2)
val swappedRDD = keyRDD.map(_.swap)
swappedRDD.foreach(x => println(x))