Calculate a moving average in Swift (Swift 4.1)

moving-average.swift

extension Collection where Element == Int, Index == Int {

    /// Calculates a moving average.
    /// - Parameter period: the period to calculate averages for.
    /// - Warning: the supplied `period` must be larger than 1.
    /// - Warning: the supplied `period` should not exceed the collection's `count`.
    /// - Returns: a dictionary of indexes and averages.
    func movingAverage(period: Int) -> [Int: Float] {
        precondition(period > 1)
        precondition(count > period)
        let result = (0..<self.count).compactMap { index -> (Int, Float)? in
            if (0..<period).contains(index) { return nil }
            let range = index - period..<index
            let sum = self[range].reduce(0, +)
            let result = Float(sum) / Float(period)

            return (index, result)
        }
        return Dictionary(uniqueKeysWithValues: result)
    }
}

Usage:

let numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
let result = numbers.movingAverage(period: 10)

let mapped = result
    .sorted { $0.key < $1.key }
    .map { $0.value }

it should have another parameter for moving index/window some thing describe in the moving Mean