Как отсортировать каждую строку rdd в искре, используя scala?
Мой текстовый файл имеет следующие данные:
10,14,16,19,52
08,09,12,20,45
55,56,70,78,53
Я хочу отсортировать каждую строку в порядке убывания. Я попробовал приведенный ниже код
val file = sc.textFile("Maximum values").map(x=>x.split(","))
val sorted = file.sortBy(x=> -x(2).toInt)
sorted.collect()
Я получил вывод ниже
[[55, 56, 70, 78, 53], [10, 14, 16, 19, 52], [08, 09, 12, 20, 45]]
Приведенный выше результат показывает, что весь список был отсортирован в порядке убывания. Но я ищу, чтобы отсортировать все значения в порядке убывания
Например
[10,14,16,19,52],[08,09,12,20,45],[55,56,70,78,53]
должно быть
[52,19,16,14,10],[45,20,12,09,08],[78,70,56,55,53]
Пожалуйста, уделите время, чтобы ответить на это. Спасибо заранее.
3 ответа
Решение
Проверь это.
val file = spark.sparkContext.textFile("in/sort.dat").map( x=> { val y = x.split(','); y.sorted.reverse.mkString(",") } )
file.collect.foreach(println)
EDIT1: как различные методы применяются к приведенному выше коду.
scala> val a = "10,14,16,19,52"
a: String = 10,14,16,19,52
scala> val b = a.split(',')
b: Array[String] = Array(10, 14, 16, 19, 52)
scala> b.sorted
res0: Array[String] = Array(10, 14, 16, 19, 52)
scala> b.sorted.reverse
res1: Array[String] = Array(52, 19, 16, 14, 10)
scala> b.sorted.reverse.mkString(",")
res2: String = 52,19,16,14,10
scala> b.sorted.reverse.mkString("*")
res3: String = 52*19*16*14*10
scala>
EDIT2:
val file = spark.sparkContext.textFile("in/sort.dat").map( x=> { val y = x.split(',').map(_.toInt); y.sorted.reverse.mkString(",") } )
file.collect.foreach(println)
Вот один из способов (не проверено)
val reverseStringOrdering = Ordering[String].reverse
val file = sc.textFile("Maximum values").map(x=>x.split(",").sorted(reverseStringOrdering))
val sorted = file.sortBy(r => r, ascending = false)
sorted.collect()
Спарк SQL способ,
import org.apache.spark.sql.functions._
val df = Seq(
("10","14","16","19","52"),
("08","09","12","20","45"),
("55","56","70","78","53")).toDF("C1", "C2","C3","C4","C5")
df.withColumn("sortedCol", sort_array(array("C1", "C2","C3","C4","C5"), false))
.select("sortedCol")
.show()
Выход
+--------------------+
| sortedCol|
+--------------------+
|[52, 19, 16, 14, 10]|
|[45, 20, 12, 09, 08]|
|[78, 70, 56, 55, 53]|
+--------------------+