swift - 使用存储在两个数组中的数字和权重计算加权平均值

Question

Stack Overflow 上已经有关于如何计算 [Int:Int] 字典的加权平均值的问题，包括使用 reduce 的奇特方法。但我现在将数字和权重存储在两个数组中。不保证这些数字是相同的。 执行此操作的Swift 方法是什么？是否可以使用像 reduce 这样的 Swift 特性？我希望有一种不使用 for 循环的方法。谢谢!

let numbers = [1, 2, 4, 3, 2]
let weights = [10, 20, 30, 15, 25]

Answer 1

没有 for 循环:

func weightedAverage(values: [Double], weights: [Double]) -> Double {
    precondition(values.count > 0 && values.count == weights.count)

    let totalWeight = weights.reduce(0.0, +)
    precondition(totalWeight > 0)

    return zip(values, weights)
            .map { $0 * $1 }
            .reduce(0.0, +) / totalWeight
}

let avg = weightedAverage(values: [1, 2, 4, 3, 2], weights: [10, 20, 30, 15, 25])
print(avg)