Swift 字典默认值

Question

我已经习惯了 Python 的 defaultdicts 的模式是一个字典，如果给定键的值没有明确设置，它会返回一个默认值。尝试在 Swift 中执行此操作有点冗长。

var dict = Dictionary<String, Array<Int>>()
let key = "foo"
var value: Array<Int>! = dict[key]
if value == nil {
    value = Array<Int>()
    dict[key] = value
}

我意识到我可以创建一个类来执行此操作，但是必须通过属性访问实际的 Dictionary 才能使用任何其他正常的 Dictionary 方法

class DefaultDictionary<A: Hashable, B> {
    let defaultFunc: () -> B
    var dict = Dictionary<A, B>()

    init(defaultFunc: () -> B) {
        self.defaultFunc = defaultFunc
    }
    subscript(key: A) -> B {
        get {
            var value: B! = dict[key]
            if value == nil {
                value = defaultFunc()
                dict[key] = value
            }
            return value
        }
        set {
            dict[key] = newValue
        }
    }
}

有更好的模式吗？

Answer 1

这在 Swift 4 中发生了变化，现在有一种方法可以读取键值或在键不存在时提供默认值。例如:

let person = ["name": "Taylor", "city": "Nashville"]
let name = person["name", default: "Anonymous"]

这在修改字典值时特别有用，因为你可以这样写代码:

var favoriteTVShows = ["Red Dwarf", "Blackadder", "Fawlty Towers", "Red Dwarf"]
var favoriteCounts = [String: Int]()

for show in favoriteTVShows {
    favoriteCounts[show, default: 0] += 1
}

我在文章 What's new in Swift 4 中介绍了此更改和其他更改.