ios - Swift 中数组的直方图

Question

我正在尝试编写一个在 Array 上运行的通用直方图函数，但我遇到了困难，因为 Type 'Element' 不符合协议(protocol) 'Hashable'.

extension Array {
    func histogram() -> [Array.Element: Int] {
        return self.reduce([Array.Element: Int]()) { (acc, key) in
                let value = (acc[key] == nil) ? 1 : (acc[key]! + 1)
                return acc.dictionaryByUpdatingKey(key: key, value: value)
        }
    }
}

其中 dictionaryByUpdatingKey(...) 改变现有字典如下:

extension Dictionary {
    func dictionaryByUpdatingKey(key: Dictionary.Key, value: Dictionary.Value) -> Dictionary {
        var mutableSelf = self
        let _ = mutableSelf.updateValue(value, forKey: key)
        return mutableSelf
    }
}

我尝试用 AnyHashable 替换 Array.Element，然后强制 key as! AnyHashable，但这看起来很困惑，返回类型最好与 Array.Element 相同，而不是 AnyHashable。

我希望按如下方式使用 Array 扩展:

let names = ["Alex", "Alex", "James"]
print(names.histogram()) // ["James": 1, "Alex": 2]

或

let numbers = [2.0, 2.0, 3.0]
print(numbers.histogram()) // [3.0: 1, 2.0: 2]

Answer 1

将通用 where 子句:where Element: Hashable 添加到您的扩展中:

extension Sequence where Element: Hashable {
    func histogram() -> [Element: Int] {
        return self.reduce([Element: Int]()) { (acc, key) in
            let value = acc[key, default: 0] + 1
            return acc.dictionaryByUpdatingKey(key: key, value: value)
        }
    }
}

我还采纳了@MartinR 的建议，即使用新的 default 值进行字典查找。

---

使用 reduce(into:_:)你可以更简单有效地做到这一点:

extension Sequence where Element: Hashable {
    func histogram() -> [Element: Int] {
        return self.reduce(into: [:]) { counts, elem in counts[elem, default: 0] += 1 }
    }
}