Swift 词典理解

Question

其他语言，例如 Python 允许您使用字典推导式从数组生成字典，但我在 Swift 中还没有弄清楚如何做到这一点。我以为我可以使用这样的东西，但它无法编译：

let x = ["a","b","c"]
let y = x.map( { ([=11=]:"x") })
// expected y to be ["a":"x", "b":"x", "c":"x"]

从 swift 中的数组生成字典的正确方法是什么？

Answer 1

map 方法简单地将数组的每个元素转换为一个新元素。然而，结果仍然是一个数组。要将数组转换为字典，您可以使用 reduce 方法。

let x = ["a","b","c"]
let y = x.reduce([String: String]()) { (var dict, arrayElem) in
    dict[arrayElem] = "this is the value for \(arrayElem)"
    return dict
}

这将生成字典

["a": "this is the value for a",
 "b": "this is the value for b",
 "c": "this is the value for c"]

一些解释：reduce的第一个参数是初始值，在本例中是空字典[String: String]()。 reduce 的第二个参数是一个回调，用于将数组的每个元素组合成当前值。在这种情况下，当前值是字典，我们在其中为每个数组元素定义一个新的键和值。修改后的字典也需要在回调中返回。

Update：由于 reduce 方法对大型数组的内存占用很大（请参阅评论），您还可以定义类似于以下代码段的自定义理解函数.

func dictionaryComprehension<T,K,V>(array: [T], map: (T) -> (key: K, value: V)?) -> [K: V] {
    var dict = [K: V]()
    for element in array {
        if let (key, value) = map(element) {
            dict[key] = value
        }
    }
    return dict
}

调用该函数如下所示。

let x = ["a","b","c"]
let y = dictionaryComprehension(x) { (element) -> (key: String, value: String)? in
    return (key: element, value: "this is the value for \(element)")
}

更新 2：除了自定义函数，您还可以在 Array 上定义扩展，这将使代码更易于重用。

extension Array {
    func toDict<K,V>(map: (T) -> (key: K, value: V)?) -> [K: V] {
        var dict = [K: V]()
        for element in self {
            if let (key, value) = map(element) {
                dict[key] = value
            }
        }
        return dict
    }
}

调用上面的代码如下所示。

let x = ["a","b","c"]
let y = x.toDict { (element) -> (key: String, value: String)? in
    return (key: element, value: "this is the value for \(element)")
}

Swift 词典理解

Swift dictionary comprehension

dictionary-comprehension

swift