参考《Spark快速大数据分析》动物书中的第四章"键值对操作",由于pair RDD的一些特殊操作,没有和前面两篇的API归纳放在一起做示例
前面的几个api —— reduceByKey()函数、foldByKey()函数、groupByKey()函数、combineByKey()函数、mapValues()函数、flatMapValues()函数、keys()函数、values()函数和sortByKey函数是针对一个Pair RDD的操作
而后的几个api —— subtractByKey()函数、join()函数、rightOuterJoin()函数、leftOuterJoin()函数和cogroup()函数是针对两个Pair RDD的函数
Spark转化操作API归纳:https://www.cnblogs.com/kuluo/p/12545374.html
Spark行动操作API归纳:https://www.cnblogs.com/kuluo/p/12550938.html
Pair RDD行动操作API归纳:https://www.cnblogs.com/kuluo/p/12567221.html
idea + spark 2.4.5 + scala 2.11.12
RDD均通过SparkContext的parallelize()函数创建
目的:
合并具有相同键的值,在经典的WordCount示例程序里,通过该函数合并相同字母(键)的出现次数(值)
代码:
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _).foreach(pair => print(s"$pair "))
**输出:
**
(d,2) (a,3) (b,5) (c,4)
**目的:
**
合并具有相同键的值,同reduceByKey()类似,不过要指定初始值
**代码:
**
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1)).foldByKey(0)(_ + _).foreach(pair => print(s"$pair "))
**输出:
**
(d,2) (a,3) (b,5) (c,4)
**目的:
**
对pair RDD中的每个值应用一个函数而不改变键
**代码:
**
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1))
.reduceByKey(_ + _)
.mapValues(_ * 2)
.foreach(pair => print(s"$pair "))
**输出:
**
(d,4) (a,6) (b,10) (c,8)
**目的:
**
对pair RDD中的每个值应用一个返回迭代器的函数,然后对返回的每个元素都生成一个对应原键的键值对记录
**代码:
**
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1))
.reduceByKey(_ + _)
.flatMapValues(_ to 5)
.foreach(pair => print(s"$pair "))
**输出:
**
(d,2) (d,3) (d,4) (d,5) (a,3) (a,4) (a,5) (b,5) (c,4) (c,5)
**目的:
**
对具有相同键的值进行分组
**代码:
**
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1))
.groupByKey()
_/**
* (d,CompactBuffer(1, 1)) (a,CompactBuffer(1, 1, 1)) (b,CompactBuffer(1, 1, 1, 1, 1)) (c,CompactBuffer(1, 1, 1, 1))
*/
.mapValues(_.sum) .foreach(pair => _print(s"$pair "))
**输出:
**
使用groupByKey()函数后,直接输出的结果为
(d,CompactBuffer(1, 1)) (a,CompactBuffer(1, 1, 1)) (b,CompactBuffer(1, 1, 1, 1, 1)) (c,CompactBuffer(1, 1, 1, 1))
在groupByKey()函数后使用mapValues()函数,可以实现和WordCount的reduceByKey()函数相同的效果,输出的结果为
(d,2) (a,3) (b,5) (c,4)
**更高效的操作:
**
可以一步到位的reduceByKey()函数效率更高,避免使用groupByKey()函数分步操作,以下为源码注释
This operation may be very expensive. If you are grouping in order to perform an aggregation (such as a sum or average) over each key, using `PairRDDFunctions.aggregateByKey` or `PairRDDFunctions.reduceByKey` will provide much better performance.
**目的:
**
使用不同的返回类型合并具有相同键的值,大多数基于键聚合的函数都是用它实现的(如reduceByKey()函数和foldByKey()函数)
遇到一个新元素时,会使用createCombiner()的函数来创建那个键对应的累加器的初始值,这一过程会在每个分区中第一次出现各个键时发生,而不是在整个RDD中第一次出现时发生
若是一个之前遇到过的键,会使用mergeValue()的函数将该键的累加器对应的当前值与这个新的值进行合并
每个分区独立处理,对于同一个键可以有多个累加器,若有多个分区对应同一个键的累加器,会使用mergeCombiners()函数将各个分区的结果进行合并
**代码(WordCount):
**
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1))
.combineByKey(x => x,
(x:Int, y:Int) => x + y,
(x:Int, y:Int) => x + y)
.foreach(pair => print(s"$pair "))
**输出:
**
(d,2) (a,3) (b,5) (c,4)
**代码(求平均):
**
/*
* (a,3) (b,5) (c,4) (d,2)
*/
val testList1 = List("a a a b b b", "b b c c c", "c d d")
/*
* (a,5) (b,4) (c,6)
*/
val testList2 = List("a a a a a b b", "b b c c c c c", "c")
val testRdd1 = sc.parallelize(testList1)
val testRdd2 = sc.parallelize(testList2)
/*
* union_取并集,构造_(a,3) (a,5) (b,5) (b,4) (c,4) (c,6) (d,2)
*/
val testRdd3 = testRdd1.flatMap(_.split(" ")).map((_, 1))
.combineByKey(x => x,
(x: Int, y: Int) => x + y,
(x: Int, y: Int) => x + y)
.union(
testRdd2.flatMap(_.split(" ")).map((_, 1))
.combineByKey(x => x,
(x: Int, y: Int) => x + y,
(x: Int, y: Int) => x + y)
)
testRdd3.combineByKey(
x => (x, 1),
(x: (Int, Int), y) => (x._1 + y, x._2 + 1),
(x: (Int, Int), y: (Int, Int)) => (x._1 + y._1, x._2 + y._2)
).mapValues(value => 1.0 * value._1 / value._2).foreach(pair => print(s"$pair "))
**输出:
**
(d,2.0) (a,4.0) (b,4.5) (c,5.0)
**目的:
**
返回一个仅包含键的RDD
**代码:
**
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1))
.reduceByKey(_ + _)
.keys
.foreach(key => print(s"$key "))
**输出:
**
d a b c
**目的:
**
返回一个仅包含值得RDD
**代码:
**
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1))
.reduceByKey(_ + _)
.values
.foreach(value => print(s"$value "))
**输出:
**
2 3 5 4
**目的:
**
返回一个根据键排序的RDD,默认升序
**代码:
**
val testList = List("a a a b b b", "b b c c c", "c d d")
val testRdd = sc.parallelize(testList)
testRdd.flatMap(_.split(" ")).map((_, 1))
.reduceByKey(_ + _)
.sortByKey(false)
.foreach(pair => print(s"$pair "))
**输出:
**
(d,2) (c,4) (b,5) (a,3)
**目的:
**
类似于集合中的差集,删掉当前RDD中键与另一个RDD中的键相同的元素
**代码:
**
/*
* (a,3) (b,5) (c,4) (d,2)
*/
val testList1 = List("a a a b b b", "b b c c c", "c d d")
/*
* (a,5) (b,4) (c,6)
*/
val testList2 = List("a a a a a b b", "b b c c c c c", "c")
val testRdd1 = sc.parallelize(testList1)
val testRdd2 = sc.parallelize(testList2)
testRdd1.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _)
.subtractByKey(testRdd2.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _))
.foreach(pair => print(s"$pair "))
**输出:
**
(d,2)
**目的:
**
类似于集合中的并集,对两个RDD进行内连接
**代码:
**
/*
* (a,3) (b,5) (c,4) (d,2)
*/
val testList1 = List("a a a b b b", "b b c c c", "c d d")
/*
* (a,5) (b,4) (c,6)
*/
val testList2 = List("a a a a a b b", "b b c c c c c", "c")
val testRdd1 = sc.parallelize(testList1)
val testRdd2 = sc.parallelize(testList2)
testRdd1.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _)
.join(testRdd2.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _))
.foreach(pair => print(s"$pair "))
**输出:
**
(a,(3,5)) (b,(5,4)) (c,(4,6))
**目的:
**
类似于集合的右连接,确保函数内的RDD的键存在
**代码:
**
/*
* (a,3) (b,5) (c,4) (d,2)
*/
val testList1 = List("a a a b b b", "b b c c c", "c d d")
/*
* (a,5) (b,4) (c,6)
*/
val testList2 = List("a a a a a b b", "b b c c c c c", "c")
val testRdd1 = sc.parallelize(testList1)
val testRdd2 = sc.parallelize(testList2)
testRdd1.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _)
.rightOuterJoin(testRdd2.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _))
.foreach(pair => print(s"$pair "))
**输出:
**
(a,(Some(3),5)) (b,(Some(5),4)) (c,(Some(4),6))
**目的:
**
类似于集合的左连接,确保调用该函数的RDD的键存在
**代码:
**
/*
* (a,3) (b,5) (c,4) (d,2)
*/
val testList1 = List("a a a b b b", "b b c c c", "c d d")
/*
* (a,5) (b,4) (c,6)
*/
val testList2 = List("a a a a a b b", "b b c c c c c", "c")
val testRdd1 = sc.parallelize(testList1)
val testRdd2 = sc.parallelize(testList2)
testRdd1.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _)
.leftOuterJoin(testRdd2.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _))
.foreach(pair => print(s"$pair "))
**输出:
**
(d,(2,None)) (a,(3,Some(5))) (b,(5,Some(4))) (c,(4,Some(6)))
**目的:
**
将两个RDD中拥有相同键的数据分组到一起
**代码:
**
/*
* (a,3) (b,5) (c,4) (d,2)
*/
val testList1 = List("a a a b b b", "b b c c c", "c d d")
/*
* (a,5) (b,4) (c,6)
*/
val testList2 = List("a a a a a b b", "b b c c c c c", "c")
val testRdd1 = sc.parallelize(testList1)
val testRdd2 = sc.parallelize(testList2)
testRdd1.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _)
.cogroup(testRdd2.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _))
.foreach(pair => print(s"$pair "))
**输出:
**
(d,(CompactBuffer(2),CompactBuffer())) (a,(CompactBuffer(3),CompactBuffer(5)))
(b,(CompactBuffer(5),CompactBuffer(4))) (c,(CompactBuffer(4),CompactBuffer(6)))
手机扫一扫
移动阅读更方便
你可能感兴趣的文章