How to count matching items in an array

Paul Hudson    February 13th 2018    @twostraws

If you want to count how many items in an array (or any collection) match a test you specify, the easiest thing to do is run the collection through a call to filter() then count the remainder.

For example, if you had an array of numbers and wanted to count how many were odd, you would write this:

let count = [1, 2, 3, 4, 5].filter { $0 % 2 == 1 }.count

Because this is something that all collections might want to do, you should consider wrapping it in an extension on Collection, like this:

extension Collection {
    func count(where test: (Element) throws -> Bool) rethrows -> Int {
        return try self.filter(test).count
    }
}

With that change, counting the odd numbers becomes this:

let count = [1, 2, 3, 4, 5].count { $0 % 2 == 1 }

Available from iOS 8.0

Did this solution work for you? Please pass it on!

Tweet

Other people are reading…

• How to use try/catch in Swift to handle exceptions
• What does override mean?
• What’s the difference between a protocol and a class?
• What is function composition?
• How to convert a string to a float

About the Swift Knowledge Base

This is part of the Swift Knowledge Base, a free, searchable collection of solutions for common iOS questions.