scala – Spark中的分区

var arr: Array[(Int,Char)] = Array()
for (i <- 'a' to 'z') {arr = arr :+ (1,i)} // Key 1 has 25 elements
for (i <- List.range('a','c')) {arr = arr :+ (2,i)} // Key 2 has 2
for (i <- List.range('a','f')) {arr = arr :+ (3,i)} // Key 3 has 5
val rdd = sc.parallelize(arr,8)

我想对上面的RDD进行分区,以便每个分区都接收不同的密钥,并且分区的大小几乎相同.下面的代码允许我按键对RDD进行分区：

val prdd = rdd.partitionBy(new HashPartitioner(3))

上述代码创建的分区具有以下大小：

scala> prdd.mapPartitions(iter=> Iterator(iter.length)).collect
 res43: Array[Int] = Array(25,2,5)

有没有办法可以从这个rdd制作几乎相同大小的分区？因此,例如对于上面的情况,键1具有最大的分区大小25.我可以具有如下分区大小：

Array[Int] = Array(5,5,5)

我尝试在上面的prdd上做RangePartition,但它没有用.