1

我有一个转换以下数据类型的代码:

从:SortedMap[Long, SortedMap[String, Double]] 到:SortedMap[String, Array[Double]]

谁能解释一下这段代码是如何进行上述操作的?

val convertedDataWindow = dataWindow.values.foldLeft(SortedMap.empty[String, Array[Double]]) {
  case (res0, map) =>
    (res0 /: map) {
      case (res1, (key, value)) =>
        res1.updated(key, res1.getOrElse(key, Array.empty) :+ value)
    }
}
4

1 回答 1

2

首先考虑内部函数:

def something(res0: SortedMap[String, Array[Double]],
  map: SortedMap[String, Double]) = (res0 /: map) {
  case (res1, (key, value)) =>
    res1.updated(key, res1.getOrElse(key, Array.empty) :+ value)
}

/:是 的另一个名称foldLeft,这与

map.foldLeft(res0) { ...}

你知道foldLeft一般情况下是如何工作的吗?它遍历集合,使用给定的函数将每个值依次“合并”为初始值。所以这从 开始res0,对于每一个(key, value)in map,我们使用 将它合并到我们的工作结果中.updated:我们向地图添加一个新条目,带有 keykey和 value res1.getOrElse(key, Array.empty) :+ value。也就是说,如果这个键已经有一个条目,我们添加value到它的末尾;否则,我们创建一个新Array的包含 just value

所以把所有这些放在一起,这个函数正在做的是合并mapres0,通过为任何新键创建新条目,或者将值放在Array任何现有键的值上。

现在,完整的功能正在做另一个foldLeft;我们可以写成:

dataWindow.values.foldLeft(SortedMap.empty)(something)

所以这是从一个空地图开始,然后依次将每个dataWindow.values地图合并到我们的工作地图中。

(当然,当我谈到“合并”时,实际上一切都是不可变的,每次循环时我们都会创建一个新的“工作”映射。但有时更清楚地想象一个单一的、可变的“工作”映射res

于 2014-12-04T09:45:48.113 回答