scala中的List列表

Question

我想知道如何在 reduce 操作的结果中创建 List 列表。

我有例如这行

1,2,3,4
0,7,8,9
1,5,6,7
0,6,5,7

我想得到这样的东西

1, [[2,3,4],[5,6,7]]
0, [[7,8,9],[6,5,7]]

这是我的代码

val parsedData = data.map { line =>
    val parts = line.split(",")
    val label = Integer.parseInt(parts(0))
    (label, List(Integer.parseInt(parts(1)), Integer.parseInt(parts(2)), Integer.parseInt(parts(3)))
}

这样我就明白了

1, [2,3,4]
0, [7,8,9]
1, [5,6,7]
0, [6,5,7]

但是，如果我将 reduceByKey 操作与 List.concat(_,_) 一起使用，我会得到一个包含所有项目的列表。

parsedData.reduceByKey(List.concat(_,_))

我想要一个 List 列表，由 Key 减少。

还有其他不知道的操作吗？

非常感谢您的帮助！

Answer 1

这是一个工作示例：

val data = "1,2,3,4\n0,7,8,9\n1,5,6,7\n0,6,5,7".split("\n")

val parsedData = data.map{ line =>
  val parts = line.split(",")
  val label = Integer.parseInt(parts(0))
  (label, List(Integer.parseInt(parts(1)), Integer.parseInt(parts(2)), Integer.parseInt(parts(3))))
}.toList
//parsedData: List[(Int, List[Int])] = List((1,List(2, 3, 4)), (0,List(7, 8, 9)), (1,List(5, 6, 7)), (0,List(6, 5, 7)))

parsedData.groupBy(_._1).mapValues(_.map(_._2))
// Map(1 -> List(List(2, 3, 4), List(5, 6, 7)), 0 -> List(List(7, 8, 9), List(6, 5, 7)))

Answer 2

我不确定这是你要找的 concat。

你可以试试吗：

parsedData.reduceByKey(_ :: _ :: Nil)

这应该从字面上创建一个包含您的元素的新列表