How to count matching items in an array

Written by Paul Hudson    @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…

• What is a lazy sequence?
• How to conform to the Hashable protocol
• What is a float?
• What’s the difference between a class and a struct?
• How to pass the Fizz Buzz test

About the Swift Knowledge Base

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