swift - Swift有没有类似numpy.diff的函数计算数组相邻元素的差值

Question

我正在尝试将一些 Python 代码转换为 Swift，并想知道是否存在一个现有函数来计算 Swift 数组中连续元素之间的差异。例如:

diff([1,3,5,6,10]) would return [2,2,1,4]

Answer 1

不，但它可以很容易地实现:

let a = [1, 3, 5, 6, 10]
zip(a.dropFirst(), a).map(-) // => [2, 2, 1, 4]

它很简单，可能不值得包装成一个函数，但如果你坚持:

extension Collection where Element: Numeric {
    func diff() -> [Element] {
        return zip(self.dropFirst(), self).map(-)
    }
}

[1, 3, 5, 6, 10].diff() // => [2, 2, 1, 4]

如果您需要延迟评估结果，您可以这样做:

extension Collection where Element: Numeric {
    func diff() -> AnyCollection<Element> {
        return AnyCollection(zip(self.dropFirst(), self).lazy.map(-))
    }
}