Scalaの一般的なコレクションのコレクション

Question

私はSeq[(A, B)]を持っています。そのようなコレクションに暗黙のメソッドを追加して、.toMultiMapを取得してMap[A, Seq[B]]に戻すことができます。

implicit class PairsExtensions[A, B](t: Traversable[(A, B)]) { 
    def toMultiMap: Map[A, Traversable[B]] = t.groupBy(_._1).mapValues(_.map(_._2)) 
    } 

しかし、今の問題は、私は常に値に対するTraversableを取り戻すです：

これが私の最初の試みでした。 Set[(A, B)].toMultiMapの場合はMap[A, Set[B]]に戻したいです。それは動作しません、

implicit class PairsExtensions2[A, B, Repr[_] <: Traversable[(A, B)]](t: TraversableLike[(A, B), Repr[(A, B)]]) { 
    def toMultiMap(implicit bf: CanBuild[B, Repr[B]]): Map[A, Repr[B]] = t.groupBy(_._1).mapValues(_.map(_._2)) 
    } 

：しかし

val m1: Map[Int, Set[String]] = Set.empty[(Int, String)] 
val m2: Map[Int, List[String]] = List.empty[(Int, String)] 

これを行う方法を教えて

だから、私はこのような何かを試してみましたか？

Answer 1

私はこれがあなたの後にあるかもしれないと思います。

は

import scala.collection.generic.CanBuildFrom 
import scala.language.higherKinds 

implicit class PairsExtensions[A, B, C[X] <: Traversable[X]](t: C[(A, B)]) { 
    def toMultiMap(implicit cbf: CanBuildFrom[Nothing, B, C[B]]): Map[A, C[B]] = 
    t.groupBy(_._1).mapValues(_.map(_._2).to[C]) 
} 

これは、いくつかの簡単なテストに合格：

scala> val m1: Map[Int, Set[String]] = Set.empty[(Int, String)].toMultiMap 
m1: Map[Int,Set[String]] = Map() 

scala> val m2: Map[Int, List[String]] = List.empty[(Int, String)].toMultiMap 
m2: Map[Int,List[String]] = Map() 

scala> Seq(('c',4),('x',2),('c',5)).toMultiMap 
res4: Map[Char,Seq[Int]] = Map(x -> Vector(2), c -> Vector(4, 5))