SparkRDD算子--partitionBy算子

语法

val newRdd = oldRdd1.partitionBy(new org.apache.spark.HashPartitioner(partitions))

partitions 表示分区数

源码

def partitionBy(partitioner : org.apache.spark.Partitioner) : org.apache.spark.rdd.RDD[scala.Tuple2[K, V]] = { /* compiled code */ }

作用

对K-V类型的RDD重新分配分区。

例子

package com.day1

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}

object oper {
    def main(args: Array[String]): Unit = {
        val config:SparkConf = new SparkConf().setMaster("local[*]").setAppName("wordCount")

        // 创建上下文对象
        val sc = new SparkContext(config)

        // partitionBy算子
        val arrayRdd = sc.makeRDD(Array((1,"张三"),(2,"李四"),(3,"王五"),(4,"刘六")),4)

        val partitionByRdd = arrayRdd.partitionBy(new org.apache.spark.HashPartitioner(2))
        println(partitionByRdd.getNumPartitions)
    }
}



输入
(1,"张三"),(2,"李四"),(3,"王五"),(4,"刘六")
输出
2

示意图

在这里插入图片描述

partitionBy算子Spark中用于重新分区的算子,它只能作用于PairRDD。当作用于PairRDD时,partitionBy的行为与repartition不同。partitionBy需要指定一个Partitioner来决定如何将每个(K,V)对根据Key值分配到对应的Partition上。默认的Partitioner是HashPartitioner,它根据Key的Hash值来计算Partition的位置。而repartition则是将数据随机打散均匀地分布在各个Partition上。 使用partitionBy算子可以在适当的情况下减少shuffle次数,提高效率。例如,可以根据某个特定的规则来重新分区数据,使得相同Key的数据在同一个Partition上,这样可以方便进行某些操作,如聚合操作。在使用partitionBy时,可以传入不同的Partitioner来指定不同的分区策略,如HashPartitioner或RangePartitioner。 以下是一个使用partitionBy的示例代码: val inputRDD = sc.parallelize(Array[(Int,Char)] ((1, 'a'),(2,'b'),(3,'C') (4,'d'),(2,'e'),(3,'f'),(2,'g'),(1, 'h')),3) val resultRDD = inputRDD.partitionBy(new HashPartitioner(2)) // 使用HashPartitioner重新分区 可以看到,在上述代码中,我们使用了HashPartitioner来将数据重新分区为2个Partition。通过这种方式,我们可以将相同Key的数据放在同一个Partition上,方便后续操作。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [[spark] repartitionpartitionBy的区别](https://blog.csdn.net/da_kao_la/article/details/106137757)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* *3* [spark-算子-分区算子partitionBy、coalesce、repartition](https://blog.csdn.net/qq_34224565/article/details/109508076)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

寒 暄

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值