swiftで数字の配列[1, 2, 3, 4, 5]などから合計と平均を出す機会が何度かあったのですが、毎回調べて実装していたので、まとめてみました。
Array.reduceを使う
数字の配列の合計を計算する場合、愚直にforEachを使う方法もありますが、Array.reduceを使うと簡潔に書けます。
[1, 2, 3, 4].reduce(0, +)
=>結果は10
第１引数は初期値（今回は０）です。
第２引数はオペレーターです。+-*/の４つが使えます。
参考
https://developer.apple.com/documentation/swift/array/2298686-reduce
extensionを使い、合計と平均を簡単に計算する
Int配列を例に、合計と平均を出すmethodを定義します。
Arrayのベース
extension Collection where Element == Int {
func sum() -> Int {
return reduce(0, +)
}
func average() -> Double {
let value = sum()
return Double(value) / Double(count)
}
}
ソース
Intだけでなく、Float, Doubleの配列についてもextensionにしたソースを以下に置きました。
https://gist.github.com/usk2000/230d46869c9eba401978a8865d2f66fa
終わりに
今回はよく使う合計と平均について書いてみましたが、もっと数学的なものも書いてみたいですね。
参考
https://qiita.com/motokiee/items/cf83b22cb34921580a52
https://developer.apple.com/documentation/swift/array/2298686-reduce