KotlinのgroupByでコレクションをグループ分けする方法【初心者向け完全ガイド】
生徒
「Kotlinでリストの中身を条件ごとにグループ分けすることってできますか?」
先生
「できますよ。そんなときに便利なのがgroupBy関数です。データを指定したルールでまとめられます。」
生徒
「ルールでまとめるってどういうことですか?」
先生
「例えばクラスの生徒を出席番号や成績で分けたり、商品のリストをカテゴリーごとにまとめたりできます。実際のコードで見ていきましょう。」
1. groupByとは?
KotlinのgroupByは、コレクション(リストや配列など)の要素を、指定した条件(キー)ごとにまとめる関数です。結果は「キー」と「そのキーに属する要素のリスト」をセットにしたMap型になります。
日常生活で例えると、本棚の本をジャンルごとに棚分けするようなものです。ジャンルがキー、本のリストが値に相当します。
2. 基本的なgroupByの使い方
まずはシンプルな例として、数字を偶数・奇数に分けるコードを見てみましょう。
fun main() {
val numbers = listOf(1, 2, 3, 4, 5, 6)
val grouped = numbers.groupBy { it % 2 == 0 }
println(grouped)
}
{false=[1, 3, 5], true=[2, 4, 6]}
ここではit % 2 == 0をキーとして、偶数(true)と奇数(false)に分けています。
3. 文字列を条件でグループ分け
次は、名前の頭文字ごとにグループ分けする例です。
fun main() {
val names = listOf("佐藤", "鈴木", "高橋", "田中", "斎藤")
val groupedByInitial = names.groupBy { it.first() }
println(groupedByInitial)
}
{佐=[佐藤], 鈴=[鈴木], 高=[高橋], 田=[田中], 斎=[斎藤]}
first()で名前の最初の文字を取得し、その文字をキーにしてまとめています。
4. データクラスとgroupByの活用
データクラスと組み合わせると、さらに実用的なデータ整理ができます。例えば、生徒の成績でグループ分けする場合です。
data class Student(val name: String, val grade: String)
fun main() {
val students = listOf(
Student("山田", "A"),
Student("佐藤", "B"),
Student("鈴木", "A"),
Student("田中", "C")
)
val groupedByGrade = students.groupBy { it.grade }
println(groupedByGrade)
}
{A=[Student(name=山田, grade=A), Student(name=鈴木, grade=A)],
B=[Student(name=佐藤, grade=B)],
C=[Student(name=田中, grade=C)]}
成績(grade)をキーとして、生徒をA・B・Cごとにまとめています。
5. 実用例:商品のカテゴリー分け
ショッピングサイトの商品をカテゴリーごとにまとめる例です。
data class Product(val name: String, val category: String)
fun main() {
val products = listOf(
Product("ノートパソコン", "家電"),
Product("冷蔵庫", "家電"),
Product("シャツ", "衣類"),
Product("ズボン", "衣類"),
Product("りんご", "食品")
)
val groupedByCategory = products.groupBy { it.category }
println(groupedByCategory)
}
{家電=[Product(name=ノートパソコン, category=家電), Product(name=冷蔵庫, category=家電)],
衣類=[Product(name=シャツ, category=衣類), Product(name=ズボン, category=衣類)],
食品=[Product(name=りんご, category=食品)]}
カテゴリー(category)をキーにして、商品のリストを整理しています。
6. groupByの注意点
- 戻り値は
Map<K, List<T>>型(キーと要素リストの組み合わせ)になります。 - 同じキーを持つ要素は自動的に同じグループに入ります。
- キーが存在しない場合、そのキーは生成されません。
このように、groupByを使えばデータ整理が簡単になります。特に集計や分類が必要な場面で威力を発揮します。
まとめ
KotlinのgroupByは、たくさんのデータを直感的に整理できる便利な仕組みでした。数字や文字列だけでなく、データクラスのような複雑な情報でも、条件をひとつ指定するだけで自然な形にまとまるため、整理や分類が必要な場面で大いに役立ちます。たとえば生徒を成績ごとにまとめたり、商品をカテゴリーで分けたり、入力された文字列を最初の文字で整理したりするような処理は、昔から多くのプログラムで必要とされてきた機能です。このような分類処理は、従来であれば何度もfor文を書いて条件を調べ、結果を手動でリストに追加しなければならず、長くて読みにくいコードになりがちでした。しかしKotlinでは、groupByを使うことで自然な文の流れのように記述でき、完成したコードも落ち着いた読みやすさを保てます。
特に見やすさと保守性という観点では、groupByの価値が際立ちます。分類処理を行うとき、複雑な条件が並ぶと後から読み返したときに迷うことがありますが、groupByなら「どの条件で仕分けているか」がひと目で分かります。条件を変更したいときも、ラムダの中身を変えるだけで済むため、書いた本人はもちろん、別の人が読んだときにも理解しやすい形式になります。システムが大きくなればなるほど、後から修正するときのわかりやすさが大きな違いにつながります。仕事や趣味の開発でも、後から読み返して整理しやすいコードは大きな武器になります。
また、groupByは単なる分類だけではありません。分類した結果はMap型として受け取るため、後から特定のキーを調べたり、グループごとの要素を利用した追加処理を行うこともできます。「家電だけ在庫を集計する」「A評価の生徒を表示する」「頭文字が同じ名前を抽出する」といった処理が自然につながるため、応用の幅が広がります。キーと値の関係がきれいにまとまることで、データの構造が頭の中で整理しやすくなり、より複雑な処理へ進むきっかけにもなります。
実際のサンプルでは、生徒を成績でまとめる例や、商品をカテゴリーで分ける例を紹介しましたが、他にも応用できる場面は多くあります。たとえば注文履歴を日付でまとめる、メッセージを送信者ごとに整理する、ログをレベルごとに仕分けるなど、さまざまなプログラムで活躍できます。数が多いデータほどgroupByの力がわかりやすく、複雑になりやすい処理が短いコードでまとめられるため、初心者でも使い心地の良さを実感できます。
さらにKotlinでは、groupByと他の標準関数を組み合わせることで、より柔軟なデータ操作が可能になります。filter、map、sortedなどと一緒に使えば、先に並び替えてから分類したり、分類後に加工したりすることもできます。単独でも強力ですが、組み合わせることで整理されたデータを別の形に変換し、集計や表示に役立てる流れが自然に作れます。短いコードで整った結果が得られるため、コードを書く人にとっても読む人にとってもうれしい形になります。
どの言語でもデータの整理は欠かせない作業であり、KotlinのgroupByはその大切な部分を支えてくれる存在です。小さなリストでも、大きなデータでも、扱うデータの種類が増えても、同じ考え方で使い続けられます。あらためて振り返ると、最初はむずかしそうに見える書き方でも、実際には一行で役割がはっきり表現でき、短くても意味が伝わる形になっていました。「この条件でまとめたい」という気持ちをそのままコードに書けるため、初心者でも取り組みやすくなります。
応用できるサンプルコード
data class Order(val item: String, val day: String)
fun main() {
val orders = listOf(
Order("りんご", "月"),
Order("シャツ", "火"),
Order("冷蔵庫", "月"),
Order("ズボン", "水"),
Order("テレビ", "火")
)
val groupedByDay = orders.groupBy { it.day }
println(groupedByDay)
}
{月=[Order(item=りんご, day=月), Order(item=冷蔵庫, day=月)],
火=[Order(item=シャツ, day=火), Order(item=テレビ, day=火)],
水=[Order(item=ズボン, day=水)]}
この短いサンプルでも、曜日ごとに注文が整理され、必要なときに特定の曜日だけ取り出すことができます。分類の考え方は生徒、生産品、文字列などどれでも同じように使えます。
生徒
「groupByって、こんなにいろいろ使えるんですね。数字も名前も商品も同じように分類できるなんて便利です。」
先生
「そうですね。条件だけ決めれば自然にデータがまとまるので、読みやすくて失敗しにくい書き方になります。後から別の条件に変えたいときも楽ですよ。」
生徒
「たくさんデータがあっても整理できるって心強いです。仕事でも使われていそうな感じがします。」
先生
「まさにその通りで、分類や集計が必要な場面ではよく使われています。覚えておくといろいろ応用できますよ。」
生徒
「今日の内容を使って、自分でも別のデータで試してみます!」