如何按swift数组分组？

func groupBy<T, K>(array: [T], keySelector: (T) -> K) -> [K: [T]] {
    var groupedDictionary = [K: [T]]()
    
    for element in array {
        let key = keySelector(element)
        
        if var elements = groupedDictionary[key] {
            elements.append(element)
            groupedDictionary[key] = elements
        } else {
            groupedDictionary[key] = [element]
        }
    }
    
    return groupedDictionary
}

let numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
let groupedNumbers = groupBy(array: numbers) { $0 % 2 == 0 ? "Even" : "Odd" }