33

キーと値のペア (String、Int) のシーケンスがあり、それらをキーごとに値のシーケンス (つまり ) にグループ化したいSeq[(String, Int)]) => Map[String, Iterable[Int]])。

明らかに、toMapここでは役に立たずgroupBy、値をタプルとして維持します。私が思いついた最高のものは次のとおりです。

val seq: Seq[( String, Int )]
// ...
seq.groupBy( _._1 ).mapValues( _.map( _._2 ) )

これを行うよりクリーンな方法はありますか？

4

4 に答える 4

19

toMultiMaptraversable にメソッドを追加する pimp を次に示します。それはあなたの問題を解決しますか？

import collection._
import mutable.Builder
import generic.CanBuildFrom

class TraversableOnceExt[CC, A](coll: CC, asTraversable: CC => TraversableOnce[A]) {

  def toMultiMap[T, U, That](implicit ev: A <:< (T, U), cbf: CanBuildFrom[CC, U, That]): immutable.Map[T, That] =
    toMultiMapBy(ev)

  def toMultiMapBy[T, U, That](f: A => (T, U))(implicit cbf: CanBuildFrom[CC, U, That]): immutable.Map[T, That] = {
    val mutMap = mutable.Map.empty[T, mutable.Builder[U, That]]
    for (x <- asTraversable(coll)) {
      val (key, value) = f(x)
      val builder = mutMap.getOrElseUpdate(key, cbf(coll))
      builder += value
    }
    val mapBuilder = immutable.Map.newBuilder[T, That]
    for ((k, v) <- mutMap)
      mapBuilder += ((k, v.result))
    mapBuilder.result
  }
}

implicit def commomExtendTraversable[A, C[A] <: TraversableOnce[A]](coll: C[A]): TraversableOnceExt[C[A], A] =
  new TraversableOnceExt[C[A], A](coll, identity)

次のように使用できます。

val map = List(1 -> 'a', 1 -> 'à', 2 -> 'b').toMultiMap
println(map)  // Map(1 -> List(a, à), 2 -> List(b))

val byFirstLetter = Set("abc", "aeiou", "cdef").toMultiMapBy(elem => (elem.head, elem))
println(byFirstLetter) // Map(c -> Set(cdef), a -> Set(abc, aeiou))

次の暗黙的な定義を追加すると、Stringや などのコレクションのようなオブジェクトでも機能しますArray。

implicit def commomExtendStringTraversable(string: String): TraversableOnceExt[String, Char] =
  new TraversableOnceExt[String, Char](string, implicitly)

implicit def commomExtendArrayTraversable[A](array: Array[A]): TraversableOnceExt[Array[A], A] =
  new TraversableOnceExt[Array[A], A](array, implicitly)

それで：

val withArrays = Array(1 -> 'a', 1 -> 'à', 2 -> 'b').toMultiMap
println(withArrays) // Map(1 -> [C@377653ae, 2 -> [C@396fe0f4)

val byLowercaseCode = "Mama".toMultiMapBy(c => (c.toLower.toInt, c))
println(byLowercaseCode) // Map(97 -> aa, 109 -> Mm)
于 2012-05-28T11:53:24.157 に答える
12

標準ライブラリにはこれを行うためのメソッドやデータ構造はなく、ソリューションは可能な限り簡潔に見えます。これを複数の場所で使用する場合は、ユーティリティ メソッドに分解することをお勧めします。

def groupTuples[A, B](seq: Seq[(A, B)]) = 
  seq groupBy (_._1) mapValues (_ map (_._2))

その後、明らかに で呼び出すだけgroupTuples(seq)です。これは、CPU クロック サイクルの点で可能な限り最も効率的ではないかもしれませんが、特に効率が悪いとは思いません。

9つのタプルのリストでJean-Philippeのソリューションに対して大まかなベンチマークを行いましたが、これはわずかに高速です. どちらも、シーケンスをマップに折りたたむよりも約 2 倍高速でした (必要な出力を得るために効果的に再実装groupByします)。

于 2012-05-28T14:30:45.527 に答える
8

あなたがそれをよりきれいだと思うかどうかはわかりません：

seq.groupBy(_._1).map { case (k,v) => (k,v.map(_._2))}
于 2012-05-28T12:36:51.617 に答える