1. aggregate函數(shù)
將每個(gè)分區(qū)里面的元素進(jìn)行聚合,然后用combine函數(shù)將每個(gè)分區(qū)的結(jié)果和初始值(zeroValue)進(jìn)行combine操作。這個(gè)函數(shù)最終返回的類型不需要和RDD中元素類型一致。
seqOp操作會(huì)聚合各分區(qū)中的元素,然后combOp操作把所有分區(qū)的聚合結(jié)果再次聚合,兩個(gè)操作的初始值都是zeroValue. seqOp的操作是遍歷分區(qū)中的所有元素(T),第一個(gè)T跟zeroValue做操作,結(jié)果再作為與第二個(gè)T做操作的zeroValue,直到遍歷完整個(gè)分區(qū)。combOp操作是把各分區(qū)聚合的結(jié)果,再聚合。aggregate函數(shù)返回一個(gè)跟RDD不同類型的值。因此,需要一個(gè)操作seqOp來把分區(qū)中的元素T合并成一個(gè)U,另外一個(gè)操作combOp把所有U聚合。
例子程序:
scala> val rdd = List(1,2,3,4,5,6,7,8,9) rdd: List[Int] = List(1, 2, 3, 4, 5, 6, 7, 8, 9) scala> rdd.par.aggregate((0,0))( (acc,number) => (acc._1 + number, acc._2 + 1), (par1,par2) => (par1._1 + par2._1, par1._2 + par2._2) ) res0: (Int, Int) = (45,9) scala> res0._1 / res0._2 res1: Int = 5
過程大概這樣:
首先,初始值是(0,0),這個(gè)值在后面2步會(huì)用到。
然后,(acc,number) => (acc._1 + number, acc._2 + 1),number即是函數(shù)定義中的T,這里即是List中的元素。所以acc._1 + number,acc._2 + 1的過程如下。
1. 0+1, 0+1
2. 1+2, 1+1
3. 3+3, 2+1
4. 6+4, 3+1
5. 10+5, 4+1
6. 15+6, 5+1
7. 21+7, 6+1
8. 28+8, 7+1
9. 36+9, 8+1
結(jié)果即是(45,9)。這里演示的是單線程計(jì)算過程,實(shí)際Spark執(zhí)行中是分布式計(jì)算,可能會(huì)把List分成多個(gè)分區(qū),假如3個(gè),p1(1,2,3,4),p2(5,6,7,8),p3(9),經(jīng)過計(jì)算各分區(qū)的的結(jié)果(10,4),(26,4),(9,1),這樣,執(zhí)行(par1,par2) =>(par1._1 + par2._1, par1._2 + par2._2)就是(10+26+9,4+4+1)即(45,9),再求平均值就簡單了。
2. aggregateByKey函數(shù):
對(duì)PairRDD中相同的Key值進(jìn)行聚合操作,在聚合過程中同樣使用了一個(gè)中立的初始值。和aggregate函數(shù)類似,aggregateByKey返回值的類型不需要和RDD中value的類型一致。因?yàn)閍ggregateByKey是對(duì)相同Key中的值進(jìn)行聚合操作,所以aggregateByKey'函數(shù)最終返回的類型還是PairRDD,對(duì)應(yīng)的結(jié)果是Key和聚合后的值,而aggregate函數(shù)直接返回的是非RDD的結(jié)果。
例子程序:
import org.apache.spark.SparkConf import org.apache.spark.SparkContext object AggregateByKeyOp { def main(args:Array[String]){ val sparkConf: SparkConf = new SparkConf().setAppName("AggregateByKey").setMaster("local") val sc: SparkContext = new SparkContext(sparkConf) val data=List((1,3),(1,2),(1,4),(2,3)) val rdd=sc.parallelize(data, 2) //合并不同partition中的值,a,b得數(shù)據(jù)類型為zeroValue的數(shù)據(jù)類型 def combOp(a:String,b:String):String={ println("combOp: "+a+"\t"+b) a+b } //合并在同一個(gè)partition中的值,a的數(shù)據(jù)類型為zeroValue的數(shù)據(jù)類型,b的數(shù)據(jù)類型為原value的數(shù)據(jù)類型 def seqOp(a:String,b:Int):String={ println("SeqOp:"+a+"\t"+b) a+b } rdd.foreach(println) //zeroValue:中立值,定義返回value的類型,并參與運(yùn)算 //seqOp:用來在同一個(gè)partition中合并值 //combOp:用來在不同partiton中合并值 val aggregateByKeyRDD=rdd.aggregateByKey("100")(seqOp, combOp) sc.stop() } }
運(yùn)行結(jié)果:
將數(shù)據(jù)拆分成兩個(gè)分區(qū)
//分區(qū)一數(shù)據(jù)
(1,3)
(1,2)
//分區(qū)二數(shù)據(jù)
(1,4)
(2,3)//分區(qū)一相同key的數(shù)據(jù)進(jìn)行合并
seq: 100 3 //(1,3)開始和中立值進(jìn)行合并 合并結(jié)果為 1003
seq: 1003 2 //(1,2)再次合并 結(jié)果為 10032//分區(qū)二相同key的數(shù)據(jù)進(jìn)行合并
seq: 100 4 //(1,4) 開始和中立值進(jìn)行合并 1004
seq: 100 3 //(2,3) 開始和中立值進(jìn)行合并 1003將兩個(gè)分區(qū)的結(jié)果進(jìn)行合并
//key為2的,只在一個(gè)分區(qū)存在,不需要合并 (2,1003)
(2,1003)//key為1的, 在兩個(gè)分區(qū)存在,并且數(shù)據(jù)類型一致,合并
comb: 10032 1004
(1,100321004)
以上就是本文的全部內(nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持服務(wù)器之家。