在确定排名时考虑具有相同值的项目

在spark中，我想计算值是如何小于或等于其他值。我试图通过排名实现这一目标，但排名产生， [1,2,2,2,3,4] -> [1,2,2,2,5,6] 而我想要的是 [1,2,2,2,3,4] -> [1,4,4,4,5,6]

我可以通过排名，按等级分组然后根据组中的项目数量修改排名值来实现此目的。但这有点效率低下。有更好的方法吗？

编辑：添加了我想要完成的最小示例

import org.apache.spark.sql.SparkSession
import org.apache.spark.sql.functions.rank
import org.apache.spark.sql.expressions.Window

object Question extends App {
val spark = SparkSession.builder.appName("Question").master("local[*]").getOrCreate()

import spark.implicits._

val win = Window.orderBy($"nums".asc)

Seq(1, 2, 2, 2, 3, 4)

.toDF("nums")
.select($"nums", rank.over(win).alias("rank"))
.as[(Int, Int)]
.groupByKey(_._2)
.mapGroups((rank, nums) => (rank, nums.toList.map(_._1)))
.map(x => (x._1 + x._2.length - 1, x._2))
.flatMap(x => x._2.map(num => (num, x._1)))
.toDF("nums", "rank")
.show(false)

}
输出：