ios - 查找具有匹配键的字典值的总和

Question

我有一个[String: [String: Int]]类型的嵌套字典，看起来像这样:

var dict = [
    "A": ["a": 234, "b": 322],
    "B": ["c": 45, "d": 443],
    "C": ["b": 50, "e": 940, "f": 9430, "a": 53029]
]

我想对其进行转换，以便将所有相同的小写键的整数加在一起以形成另一个字典:

["a": 53263, "b": 382, "c": 45, "d": 50, "e": 940, "f": 9430]

我不在乎 "A", "B", "C"键。
首先，我映射了 dict以提取嵌套字典:

let result1 = dict.map { $1 }
// [["b": 322, "a": 234], ["c": 45, "d": 443], ["e": 940, "b": 50, "a": 53029, "f": 9430]]

我用 flatMap来获取元组:

let result2 = dict.map { $1 }.flatMap { $0.map { $0 }}
// [(key: "c", value: 45), (key: "d", value: 443), (key: "f", value: 9430), (key: "e", value: 940), (key: "a", value: 53029), (key: "b", value: 50), (key: "a", value: 234), (key: "b", value: 322)]

最后，我尝试使用 result2并在每个项目上进行映射，以添加相同键的所有值，并返回 [String: Int]类型的字典:

var tempArray = [String]()
var finalDict = [String: Int]()
var count = 0
let result3 = dict.map { $1 }.flatMap { $0.map { el -> [String: Int] in
    if tempArray.contains(el.key) {
        if let newCount = finalDict[el.key] {
            count += newCount
            finalDict[el.key] = newCount
        }
    } else {
        tempArray.append(el.key)
        finalDict = [el.key: el.value]
    }

    return finalDict
}}

但是，我得到的结果是:

[["b": 322], ["a": 234], ["c": 45], ["d": 443], ["d": 443], ["d": 443], ["f": 9430], ["e": 940]]

显示相同键的倍数。

Answer 1

您可以使用flatMap创建一个元组数组，然后使用reduce(into:)创建一个包含键和每个键值的总和的字典。

let result = dict.values
    .flatMap {$0}
    .reduce(into: [:]) { $0[$1.key, default: 0] += $1.value }