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…

• Using stride() to loop over a range of numbers
• How to pass the Fizz Buzz test
• What is Automatic Reference Counting (ARC)?
• What does the open keyword do?
• How to create an array by repeating an item

About the Swift Knowledge Base

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