假设我有一个列表:
[(A, a), (A, b), (A, c), (B, a), (B, d)]
我该如何将列表变成:
[(A, [a,b,c]), (B, [a,d])]
只使用一个函数吗?
谢谢
假设我有一个列表:
[(A, a), (A, b), (A, c), (B, a), (B, d)]
[(A, [a,b,c]), (B, [a,d])]
只使用一个函数吗?
谢谢
groupBy 函数可以实现此功能:
scala> val list = List((1, 'a'), (1, 'b'), (1, 'c'), (2, 'a'), (2, 'd'))
list: List[(Int, Char)] = List((1,a), (1,b), (1,c), (2,a), (2,d))
scala> list.groupBy(_._1) // grouping by the first item in the tuple
res0: scala.collection.immutable.Map[Int,List[(Int, Char)]] = Map(2 -> List((2,a), (2,d)), 1 -> List((1,a), (1,b), (1,c)))
仅仅使用groupBy方法并不能得到你所期望的格式,因此我建议你为此编写一个自定义方法。
def groupTuples[A,B](seq: Seq[(A,B)]): List[(A, List[B])] = {
seq.
groupBy(_._1).
mapValues(_.map(_._2).toList).toList
}
然后调用它以获得所需的结果。
val t = Seq((1,"I"),(1,"AM"),(1, "Koby"),(2,"UP"),(2,"UP"),(2,"AND"),(2,"AWAY"))
groupTuples[Int, String](t)
Map
。操作请求一个 List
。 - jwvh
groupBy
可以帮助你入门,但没有单一的标准库方法可以完成这个完整的转换。 - jwvh