spark算子介绍


在Spark中,算子(Operator)是对数据集(RDD、DataFrame、Dataset)进行操作的核心功能,分为两大类: 转换算子(Transformation)行动算子(Action)。以下是详细的分类和常用算子的介绍。

1. 转换算子(Transformation)

  • 定义:转换算子是惰性求值的,调用后不会立即执行,而是生成一个新的RDD、DataFrame或Dataset,并记录依赖关系,直到遇到Action算子时才触发计算
  • 特点
    • 返回一个新的RDD、DataFrame或Dataset。
    • 是惰性求值的(Lazy Evaluation)。

1.1 常用转换算子

以下是常用的转换算子及其功能:

算子功能示例
map对每个元素进行映射,返回一个新的RDDrdd.map(x => x * 2)
flatMap类似map,但可以返回多个元素,通常用于拆分操作rdd.flatMap(x => x.split(" "))
filter过滤数据,返回满足条件的元素rdd.filter(x => x > 10)
distinct去重操作,返回不重复的元素rdd.distinct()
union合并两个RDD,返回并集rdd1.union(rdd2)
intersection返回两个RDD的交集rdd1.intersection(rdd2)
subtract返回两个RDD的差集rdd1.subtract(rdd2)
cartesian返回两个RDD的笛卡尔积rdd1.cartesian(rdd2)
groupByKey按Key分组,返回一个(Key, Iterable[Value])的RDDrdd.groupByKey()
reduceByKey按Key聚合,返回一个(Key, Value)的RDDrdd.reduceByKey((x, y) => x + y)
sortByKey按Key排序rdd.sortByKey(ascending = true)
join对两个RDD进行内连接,返回(Key, (Value1, Value2))rdd1.join(rdd2)
cogroup对两个RDD按Key分组,返回(Key, (Iterable[Value1], Iterable[Value2]))rdd1.cogroup(rdd2)
mapValues(Key, Value)中的Value进行映射rdd.mapValues(value => value * 2)
flatMapValues(Key, Value)中的Value进行映射,返回多个Valuerdd.flatMapValues(value => value.split(","))
partitionBy对RDD重新分区rdd.partitionBy(new HashPartitioner(4))
coalesce减少分区数,用于优化性能rdd.coalesce(2)
repartition增加或减少分区数,类似coalesce,但会触发shufflerdd.repartition(4)
pipe调用外部脚本处理RDD中的数据rdd.pipe("script.sh")

2. 行动算子(Action)

  • 定义:行动算子会触发Spark的计算将数据从Executor返回到Driver,或者将结果输出到存储系统
  • 特点
    • 会触发实际的计算。
    • 返回非RDD类型的结果(如值、集合),或者将数据输出到外部存储

2.1 常用行动算子

以下是常用的行动算子及其功能:

算子功能示例
collect将RDD中的所有数据收集到Driver端,返回一个数组rdd.collect()
count返回RDD中元素的总数rdd.count()
first返回RDD中的第一个元素rdd.first()
take返回RDD中的前N个元素rdd.take(5)
takeSample随机采样返回RDD中的N个元素rdd.takeSample(withReplacement = false, num = 5)
takeOrdered返回RDD中排序后的前N个元素rdd.takeOrdered(5)
reduce对RDD中的元素进行归约操作rdd.reduce((x, y) => x + y)
fold类似reduce,但需要提供初始值rdd.fold(0)((x, y) => x + y)
aggregate对RDD中的元素进行聚合操作,支持分区内和分区间的聚合逻辑rdd.aggregate(0)(_ + _, _ + _)
countByKey(Key, Value)形式的RDD按Key计数rdd.countByKey()
foreach对RDD中的每个元素执行指定操作(通常用于副作用,如打印日志)rdd.foreach(println)
saveAsTextFile将RDD保存为文本文件rdd.saveAsTextFile("output_path")
saveAsSequenceFile将RDD保存为Hadoop的SequenceFile格式rdd.saveAsSequenceFile("output_path")
saveAsObjectFile将RDD保存为对象文件rdd.saveAsObjectFile("output_path")

3. 转换算子与行动算子的区别

维度转换算子(Transformation)行动算子(Action)
定义对RDD进行转换,生成新的RDD触发实际计算,返回非RDD结果或输出数据
执行时机惰性求值,调用时不会立即执行调用时立即触发计算
返回值返回RDD返回非RDD类型的结果或无返回值
示例mapfilterreduceByKeycollectcountsaveAsTextFile

4. 示例代码

以下是一个完整的示例,结合转换算子和行动算子:

val rdd = sc.textFile("data.txt")  // 读取文本文件
val words = rdd.flatMap(_.split(" "))  // 转换算子:拆分单词
val filteredWords = words.filter(word => word.length > 3)  // 转换算子:过滤长度大于3的单词
val wordPairs = filteredWords.map(word => (word, 1))  // 转换算子:转换为键值对
val wordCounts = wordPairs.reduceByKey(_ + _)  // 转换算子:按Key聚合
wordCounts.foreach(println)  // 行动算子:打印结果

5. 总结

  • 转换算子(Transformation)

    • 用于定义数据的处理逻辑,生成新的RDD。
    • 是惰性求值的,只有在调用行动算子时才会执行。
    • 常见的有:mapfilterreduceByKeyjoin等。
  • 行动算子(Action)

    • 用于触发计算,将结果返回到Driver端或输出到存储。
    • 常见的有:collectcountsaveAsTextFileforeach等。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值