如何对列表中的整数数量和浮点数求和 - Scala

Question

我需要计算我在 Map 中的整数和浮点数，例如 Map[String, List[(Int, String, Float)]]

数据来自读取文件-例如里面的数据看起来有点像（但是还有更多路由）：

Cycle Route (City),1:City Centre :0.75f,2:Main Park :3.8f,3:Central Station:2.7f,4:Modern Art Museum,5:Garden Centre:2.4f,6:Music Centre:3.4f

地图被拆分，因此 String 是路线的名称，而 List 是其余的数据。

我希望它计算每条路线的“检查点”数量和每条路线的总距离（即浮动），然后打印出例如Oor Wullie Route 有 6 个检查站，总距离 18.45km

我猜我需要使用foldLeft，但是我不确定该怎么做？

我之前做过的简单折叠示例，但不确定如何将其应用于上述场景？

val list1 = List.range(1,20)

def sum(ls:List[Int]):Int = {
  ls.foldLeft(0) { _ + _}
}

Answer 1

你可以通过折叠来做到这一点，但 IMO 是不必要的。

您只需获取列表的大小即可知道检查点的数量（假设列表中的每个条目都是一个检查点）。

要计算总距离，您可以这样做：

def getDistance(list: List[(Int, String, Float)]): Float = 
  list
    .iterator // mapping the iterator to avoid building an intermediate List instance
    .map(_._3) // get the distance Float from the tuple
    .sum // built-in method for collections of Numeric elements (e.g. Float)

然后得到你的打印输出：

def summarize(routes: Map[String, List[(Int, String, Float)]]): Unit =
  for { (name, stops) <- routes } {
    val numStops = stops.size
    val distance = getDistance(stops)
    println(s"$name has $numStops stops and total distance of $distance km")
  }

如果您真的想通过foldLeft 计算numStops 和distance，Luis 对您的问题的评论就是这样做的方法。

编辑 - 根据 Luis 的要求，将他的评论放在这里并稍微清理一下：

stops.foldLeft(0 -> 0.0f) { 
   // note: "acc" is short for "accumulated"
   case ((accCount, accDistance), (_, _, distance)) => 
     (accCount + 1) -> (accDistance + distance) 
}