spark 的一些常用函数 filter，map，flatMap，lookup ，reduce，groupByKey

定义不带参数也不带返回值的函数（def :定义函数的关键字  printz:方法名称）

scala> def printz = print("scala hello")

 

定义带参数也带返回值的函数（这种函数在定义时也可以不带返回值的类型，scala会自动推算出。建议还是带上）

scala> def minNum(x:Int,y:Int):Int = if(x>y) x else y    //:Int 是该函数的返回值类型
minNum: (x: Int, y: Int)Int

调用函数

scala> minNum(,)
res10: Int = 

 

 

创建一个map

scala> val colors = Map("red" -> "#FF0000", "azure" -> "#F0FFFF")
colors: scala.collection.immutable.Map[String,String] = Map(red -> #FF0000, azur
e -> #F0FFFF)

将map中的每一个元素中的第一个元组与第二个元组交换位置

'_'：表示map的元组，

x._2：map中每一个元素的第二个元组（#FF0000,"#F0FFFF"）

scala> colors.map(x=>(x._2,x._1))
res0: scala.collection.immutable.Map[String,String] = Map(#FF0000 -> red, #F0FFF
F -> azure)

 

union(合并rdd)

scala> var rdd1=sc.parallelize(List(,,,))//创建rdd1
rdd1: org.apache.spark.rdd.RDD[Int] = ParallelCollectionRDD[] at parallelize at <console>:
scala> var rdd2=sc.parallelize(List(,,,))//创建rdd2
rdd2: org.apache.spark.rdd.RDD[Int] = ParallelCollectionRDD[] at parallelize at <console>:
scala> var unionRes=rdd1 union rdd2  //合并rdd
unionRes: org.apache.spark.rdd.RDD[Int] = UnionRDD[] at union at <console>:
scala> unionRes.collect　　　　　  　//查合并后的rdd
res19: Array[Int] = Array(, , , , , , , )

 

join 类似于sql中的内连，左外链接，右外链接

创建rdd

scala> var rdd1=sc.parallelize(List((,"spark"),(,"hadoop"),(,"scala"),(,"java"))) //创建rdd1
rdd1: org.apache.spark.rdd.RDD[(Int, String)] = ParallelCollectionRDD[] at parallelize at <console>:
scala> var rdd2=sc.parallelize(List((,"30k"),(,"28k"),(,"5k"),(,"10k")))　　　　　　//创建rddd2
rdd2: org.apache.spark.rdd.RDD[(Int, String)] = ParallelCollectionRDD[] at parallelize at <console>:
scala> var joinRes=rdd1 join rdd2  　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　//join rdd
joinRes: org.apache.spark.rdd.RDD[(Int, (String, String))] = MapPartitionsRDD[] at join at <console>:
scala> joinRes.collect　　　//查看join后的结果，可以看出join会把两个rdd有相同key的元素的值进行合并
res20: Array[(Int, (String, String))] = Array((,(spark,30k)), (,(scala,5k)), (,(hadoop,28k)))

 

因为第4和第5没有匹配项，所以不显示

 

leftOuterJoin（左外链接）

//这个是以左边的rdd为主，右边为辅。右边与左边没有匹配项，则会显示左边的值，所以这里key为4的也会显示
scala> var leftJoinRes=rdd1.leftOuterJoin(rdd2)
leftJoinRes: org.apache.spark.rdd.RDD[(Int, (String, Option[String]))] = MapPartitionsRDD[] at leftOuterJoin at <console>:
scala> leftJoinRes.collect
res21: Array[(Int, (String, Option[String]))] = Array((,(java,None)), (,(spark,Some(30k))), (,(scala,Some(5k))), (,(hadoop,Some(28k))))
 

rightOuterJoin(右外链接)

//这个是以右边边的rdd为主，左边边为辅。右边与左边没有匹配项，则只显示右边,所以这里key为5的也会显示
scala> var rightJoinRes=rdd1.rightOuterJoin(rdd2)
rightJoinRes: org.apache.spark.rdd.RDD[(Int, (Option[String], String))] = MapPartitionsRDD[] at rightOuterJoin at <console>:
scala> rightJoinRes.collect
res22: Array[(Int, (Option[String], String))] = Array((,(Some(spark),30k)), (,(Some(scala),5k)), (,(None,10k)), (,(Some(hadoop),28k)))
 

 

groupByKey

//(4,1), (7,4), (6,3), (2,2), (3,3), (1,3)

scala> f1.flatMap(x=>x.split("-")).map((_,)).groupByKey.collect
res28: Array[(String, Iterable[Int])] = Array((,CompactBuffer()), (,CompactBuffer(, , , )), (,CompactBuffer(, , )), (,CompactBuffer(, )), (,CompactBuffer(, , )), (,CompactBuffer(, , )))

 

reduce

scala> var rdd1=sc.parallelize(List(,,,,))
rdd1: org.apache.spark.rdd.RDD[Int] = ParallelCollectionRDD[] at parallelize at <console>:
scala> rdd1.reduce(_+_)
res32: Int = 

reduce(_+_):每个元组相加 1+2+3+4+5=15

他是每次相加两个元组然后产生新的rdd与下一位元组相加

1+2=3

3+3=6

6+4=10

10+5=15

 

lookup 

scala> var rdd1=sc.parallelize(List((,"a"),(,"b"),(,"c")))
rdd1: org.apache.spark.rdd.RDD[(Int, String)] = ParallelCollectionRDD[] at parallelize at <console>:
scala> rdd1.lookup()
res34: Seq[String] = WrappedArray(a)

lookup函数对<key,value>型的rdd操作，返回指定key对应的元素形成的seq，这个函数的优点

如果这个rdd包含分区器，那么只扫描对应key所在的分区，然后返回对应key的元素形成的seq；如果这个rdd没有分区器，则对这个rdd进行全盘扫描，然后返回对应key的元素形成的seq

 

map和flatMap

map函数是以一行数据为一个元素

Array[Student] =
Array(Student(zhangxs,,chenxy), Student(wangYr,,teacher), Student(wangx,,teacher))

map是对RDD中的每个元素都执行一个指定的函数来产生一个新的RDD。任何原RDD中的元素在新RDD中都有且只有一个元素与之对应。

scala> val a = sc.parallelize( to , )
scala> val b = a.map(x => x*)
scala> a.collect
res10: Array[Int] = Array(, , , , , , , , )
scala> b.collect
res11: Array[Int] = Array(, , , , , , , , )

上述例子中把原RDD中每个元素都乘以2来产生一个新的RDD。

 

 

flatMap是一个文件的数据为一个元素

res9: Array[String] = Array(zhangxs, , chenxy, wangYr, , teacher, wangx, , teacher)

mapPartitions

mapPartitions是map的一个变种。map的输入函数是应用于RDD中每个元素，而mapPartitions的输入函数是应用于每个分区，也就是把每个分区中的内容作为整体来处理的。 
它的函数定义为：

1. def mapPartitions[U: ClassTag](f: Iterator[T] => Iterator[U], preservesPartitioning: Boolean = false): RDD[U]

f即为输入函数，它处理每个分区里面的内容。每个分区中的内容将以Iterator[T]传递给输入函数f，f的输出结果是Iterator[U]。最终的RDD由所有分区经过输入函数处理后的结果合并起来的。

比如

scala> val a = sc.parallelize( to , )
scala> def myfunc[T](iter: Iterator[T]) : Iterator[(T, T)] = {
var res = List[(T, T)]()
var pre = iter.next while (iter.hasNext) {
val cur = iter.next;
res .::= (pre, cur) pre = cur;
}
res.iterator
}
scala> a.mapPartitions(myfunc).collect
res0: Array[(Int, Int)] = Array((,), (,), (,), (,), (,), (,))

上述例子中的函数myfunc是把分区中一个元素和它的下一个元素组成一个Tuple。因为分区中最后一个元素没有下一个元素了，所以(3,4)和(6,7)不在结果中。 
mapPartitions还有些变种，比如mapPartitionsWithContext，它能把处理过程中的一些状态信息传递给用户指定的输入函数。还有mapPartitionsWithIndex，它能把分区的index传递给用户指定的输入函数。

 

mapValues

mapValues顾名思义就是输入函数应用于RDD中Kev-Value的Value，原RDD中的Key保持不变，与新的Value一起组成新的RDD中的元素。因此，该函数只适用于元素为KV对的RDD。

scala> val a = sc.parallelize(List("dog", "tiger", "lion", "cat", "panther", " eagle"), )
scala> val b = a.map(x => (x.length, x))　　//key：元组的长度 value：元组的值
scala> b.mapValues("x" + _ + "x").collect //“_”就是每个元组的value，在每个value的前后加"x"
res5: Array[(Int, String)] = Array((,xdogx), (,xtigerx), (,xlionx),(,xcatx), (,xpantherx), (,xeaglex))

mapWith

mapWith是map的另外一个变种，map只需要一个输入函数，而mapWith有两个输入函数。它的定义如下：

def mapWith[A: ClassTag, U: ](constructA: Int => A, preservesPartitioning: Boolean = false)(f: (T, A) => U): RDD[U]

• 第一个函数constructA是把RDD的partition index（index从0开始）作为输入，输出为新类型A；
• 第二个函数f是把二元组(T, A)作为输入（其中T为原RDD中的元素，A为第一个函数的输出），输出类型为U。

比如把partition index 乘以10，然后加上2作为新的RDD的元素。

val x = sc.parallelize(List(,,,,,,,,,), )
x.mapWith(a => a * )((a, b) => (b + )).collect
res4: Array[Int] = Array(, , , , , , , , , )