scala - 从scala中的文件中读取列表集合

Question

我想阅读一个包含两列整数的大文本文件。对于第一列不变的每个连续拉伸，我想将第二列收集为列表。例如，以下示例

1 2
1 1
1 2
4 3
4 8
1 5
8 2
8 2
8 7

应该产生列表

2,1,2
3,8
5
2,2,7

在 Scala 中实现这一目标的正确方法是什么？

特别是，如果有一个“惰性”解决方案会非常好，这样我就可以处理每个列表，而不必先将整个文件加载到内存中。

Answer 1

最简单的方法是使用scala.io.Source逐行读取文件。使用getLines，您可以检索一个Iterator[String]可以映射以拆分行并将它们转换为整数的值，如下所示：

val intPairs = Source.fromFile("/path/to/file").getLines.map { line =>
  line.split(" ").take(2).map(_.toInt)
}

我将连续行的分组留给你作为练习。

Answer 2

val nums = """(\d+) (\d+)""".r

val tuples = io.Source.fromFile("list.txt").getLines collect {
  case nums(label, num) => (label.toInt -> num.toInt)
}

def toList(tuples: Iterator[(Int, Int)]): Iterator[Seq[Int]] = {
  if(!tuples.hasNext) Iterator.empty
  else {
    val (label, num) = tuples.next
    val (succ, rest) = tuples.span(_._1 == label)
    Iterator(num :: succ.map(_._2).toList) ++ toList(rest)
  }
}

toList(tuples)foreach(println)