排序RDD根据在一个列中的不同的值

Question

我有一个RDD [（中间体，阵列（双））]等

1, Array(2.0,5.0,6.3) 
5, Array(1.0,3.3,9.5) 
1, Array(5.0,4.2,3.1) 
2, Array(9.6,6.3,2.3) 
1, Array(8.5,2.5,1.2) 
5, Array(6.0,2.4,7.8) 
2, Array(7.8,9.1,4.2) 

我想根据在第一列中的唯一值的RDD进行排序（ 1,5,2）

所需输出

1, Array(2.0,5.0,6.3) 
1, Array(5.0,4.2,3.1) 
1, Array(8.5,2.5,1.2) 
5, Array(1.0,3.3,9.5) 
5, Array(6.0,2.4,7.8) 
2, Array(9.6,6.3,2.3) 
2, Array(7.8,9.1,4.2) 

我与命令试图像

rdd.groupby() 
rdd.sortby() 

所有这些东西都会产生输出，排序列表像

1, Array(2.0,5.0,6.3) 
1, Array(5.0,4.2,3.1) 
1, Array(8.5,2.5,1.2) 
2, Array(9.6,6.3,2.3) 
2, Array(7.8,9.1,4.2) 
5, Array(1.0,3.3,9.5) 
5, Array(6.0,2.4,7.8) 

我如何排序具有鲜明的价值RDD是第一列由

(1,5,2) 

Answer 1

你可以先定义排序为您例如：

val ordering = (1,5,2).productIterator.toList.zipWithIndex.toMap 

，然后应用它：

rdd.sortBy{case (k,v) => ordering(k)}