我有一个转换以下数据类型的代码:
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)
}
}
答案 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
)