Swift filter, map, reduceメソッド

　こんにちは、Hiroです。久しぶりにiOSアプリをやることになり四苦八苦しています。
　さて、本日はRuby on Railsでは馴染み深いメソッドのSwift版を紹介したいと思います。

配列（Array）のfilter, map, reduce

配列に関して便利メソッドを3種類紹介します。先ずは、以下の例で操作対象となる配列を宣言しておきます。

// Array
var numbers: [Int] = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

filter

クロージャで指定した条件を満たす、新しい配列を返却します。
例として、2の倍数を取得する処理を記述します。

let evenNumbers: [Int] = numbers.filter { (value: Int) -> Bool in
    return value % 2 == 0
}
print(evenNumbers)
// => [0, 2, 4, 6, 8]

上記のようにクロージャを渡すのですが、第一引数に「$0」でアクセスでき、下記のように簡略にすることもできます。

let evenNumbers: [Int] = numbers.filter { $0 % 2 == 0 }
print(evenNumbers)
// => [0, 2, 4, 6, 8]

map

配列の１つ１つにクロージャで指定した処理を行った配列の結果を返却します。
例として、１つ１つの要素を2倍にする処理を記述します。

let doubleNumbers: [Int] = numbers.map { (value: Int) -> Int in
    return value * 2
}
print(doubleNumbers)
// => [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

簡略式も下記に記載します。

let doubleNumbers: [Int] = numbers.map { $0 * 2 }
print(doubleNumbers)
// => [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

reduce

要素のたたみ込みをするためのメソッドです。要素をまとめて一つにする場合などで利用できます。 例として、要素の合計値を取得する処理を記述します。

let totalNumber = numbers.reduce(0) { (total: Int, value: Int) -> Int in
    total + value
}
print(totalNumber)
// => 45

簡略式も下記に記載します。

let totalNumber = numbers.reduce(0) { $0 + $1 }
print(totalNumber)
// => 45

辞書（Dictionary）のfilter, map, reduce

辞書に関しても上記の「filter」「map」「reduce」のメソッドは使えますので、簡略式ですが、こちらも記載しておきます。先ずは、例で操作対象となる辞書を宣言しておきます。

// Dictionary
var dictNumbers = ["zero":0, "one": 1, "two": 2, "three": 3 , "four": 4, "five": 5, "six": 6, "seven": 7, "eight": 8, "nine": 9]

filter

Dictionaryのため、順序保証がないのと、返却値はタプルの配列になっています。

let evenDictNumbers = dictNumbers.filter { $0.1 % 2 == 0 }
print(evenDictNumbers)
// => [("four", 4), ("zero", 0), ("two", 2)] 

map

こちらもタプルとしていますが、辞書で返却したい場合が普通のケースですので、そん場合、次で紹介する「reduce」のやり方がよいでしょう。

let doubleTupleNumbers = dictNumbers.map { ($0.0, $0.1 * 2) }
print(doubleTupleNumbers)
// => [("three", 6), ("one", 2), ("nine", 18), ("seven", 14), ("four", 8), ("five", 10), ("zero", 0), ("six", 12), ("eight", 16), ("two", 4)]

reduce

let doubleDictNumbers = dictNumbers.reduce([String: Int]()) { var varDict = $0; varDict[$1.0] = $1.1 * 2; return varDict}
print(doubleDictNumbers)
// => ["three": 6, "one": 2, "nine": 18, "seven": 14, "four": 8, "five": 10, "zero": 0, "six": 12, "eight": 16, "two": 4]

上記で、注意点としては、一度「$0」を「varDict」変数に代入しているところです。これは、「$0」がイミュータブルな変数だからです。
そのため、下記ではエラーとなります。

let doubleDictNumbers = dictNumbers.reduce([String: Int]()) { $0[$1.0] = $1.1 * 2; return $0}
// =>Cannot assign through subscript: '$0' is immutable

いかがでしたでしょうか。Ruby on Railsではおなじみのメソッドですが、少しSwiftならではの制約があります。ただ、とても可読性が高く、便利なメソッドなのでどんどん利用してみようと思います。