scala - 在 Scala 中对元组列表使用 groupBy

Question

我试图在 Scala 中对元组列表进行分组。

输入:

val a = List((1,"a"), (2,"b"), (3,"c"), (1,"A"), (2,"B"))

我申请了:

a.groupBy(e => e._1)

我得到的输出是:

Map[Int,List[(Int, String)]] = Map(2 -> List((2,b), (2,B)), 1 -> List((1,a), (1,A)), 3 -> List((3,c)))

这与我的预期略有不同:

Map[Int,List[(Int, String)]] = Map(2 -> List(b, B), 1 -> List(a, A)), 3 -> List(c))

我该怎么做才能获得预期的输出？

Answer 1

你的意思可能是这样的:

a.groupBy(_._1).mapValues(_.map(_._2))

或:

a.groupBy(_._1).mapValues(_.unzip._2)

结果:

Map(2 -> List(b, B), 1 -> List(a, A), 3 -> List(c))