Dateオブジェクトの入った配列を年月でグループ分けしたい

Question

例えば下記のようなDateオブジェクトの入った配列があるとき、
年月でグループ分けするにはどのように書くと良いでしょうか？

struct MediumEntity {
    let createdAt: Date
}

var media: [MediumEntity]
let dateString = ["2016/12/1", "2017/1/1", "2017/1/2", "2017/1/3", "2017/4/1", "2017/4/1"]
let formatter = DateFormatter()
formatter.dateFormat = "yyyy/MM/dd"
media = dateString.map { MediumEntity.init(createdAt: formatter.date(from: $0)!) }

/* 下記3つのグループに分けたい
- 2016/12 （1点）
- 2017/1 （3点）
- 2017/4 （2点）
*/

mediaの範囲を求めて毎月なめていくとできそうではあるのですが。

let fromDate = media.map { $0.createdAt }.min()
let toDate = media.map { $0.createdAt }.max()

良い書き方がありましたらご教授いただけたらと思います。
よろしくお願いいたします。

Answer 1

Swift4からDictionaryのgroupingが追加されています。
【Swift】Dictionary(grouping by )を使ってセクション分けしたTableViewを作る
の記事を参考に作成できると思います。
上記の記事をコピペしてschedulesを作成するところで"yyyy/MM/dd"を"yyyy/MM"に変えたら年月でグルーピングできました。

Comments

[sabosan]

ありがとうございます！！
教えていただいた通りにやったところグルーピングできました！！

struct MediumEntity {
let createdAt: Date
}

var media: [MediumEntity]
let dateString = ["2016/12/1", "2017/1/1", "2017/1/2", "2017/1/3", "2017/4/1", "2017/4/1"]
let formatter = DateFormatter()
formatter.dateFormat = "yyyy/MM/dd"
media = dateString.map { MediumEntity.init(createdAt: formatter.date(from: $0)!) }

var classifyedMedia = [Date: [MediumEntity]]()
classifyedMedia = Dictionary(grouping: media) { medium -> Date in
formatter.dateFormat = "yyyy/MM"
let string = formatter.string(from: medium.createdAt)
return formatter.date(from: string)!
}

[nakasho_dev]

グルーピングできたようで、良かったです。出来上がったソースはコメント欄ではなく質問文の最後に追加する形のほうが、後から見た人がソースを見やすいと思います。

Answer 2

Swiftには今のところ標準で用意されたgourpBy関数がありません。
また、これが決定打だというものもないので、Swift groupByで検索して気に入ったものを使うのが良いと思います。
RubyやPythonのように高階関数になっていて要素に対応したキーを返す関数を渡すとキーごとにグループ化されたDictionayが返されるものが多いようです。

Comments

[sabosan]

ありがとうございます！！試してみます！感謝ですmm

[MasakiHori]

大変申し訳ないです。リファレンスなどから見落としていました。　nakasho_dev さんの方法が現状一番良いように思われます。

[sabosan]

いえいえ！ groupByというワードを知らなかったので勉強になりました！
ありがとうございましたmm