使用foldleft进行SortedMap操作

时间:2014-12-04 08:58:38

标签: scala map foldleft

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

from:SortedMap [Long,SortedMap [String,Double]] to: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)
    }
}

1 个答案:

答案 0 :(得分:1)

首先考虑内部函数:

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的工作原理吗?它通过集合运行,使用给定的函数来" merge"每个值依次转换为初始值。所以这从res0开始,对于(key, value)中的每个map,我们使用.updated将其合并到我们的工作结果中:我们向地图添加一个新条目,带有密钥key和价值res1.getOrElse(key, Array.empty) :+ value。也就是说,如果已经存在此密钥的条目,我们会在其末尾添加value;否则,我们会制作一个仅包含Array的新value

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

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

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

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

(当然,当我谈到"合并"时,实际上一切都是不可改变的,我们每次进行循环时都会创建一个新的"工作"地图。但有时可以想象一个单一的,可变的"工作"地图res