在 pyspark 中，我有一个如下所示的数据框，其中根据 id 和 k1 的值对行进行排序。此外，每一行都有一个唯一的升序编号（rowid）。
-----------------------
rowid | id | k1 | k2 |
-----------------------
1 | 1 | v1 | l1 |
2 | 1 | v1 | v1 |
3 | 1 | v1 | l2 |
4 | 2 | v2 | v2 |
5 | 2 | v2 | l3 |
6 | 3 | v3 | l3 |
----------------------
对于id的每个唯一值，我想计算k1==k2的第一行的rowid与观察到该id的记录的第一行对应的rowid的差+1，并存储结果在新列中（即排名）。输出应如下所示。
----------------
id | k1 |rank |
-----------------
1 | v1 | 2 | …