在Scala中添加整数元组列表

时间:2013-09-16 19:26:23

标签: scala shapeless

我希望添加一个整数元组列表,即给出一个arity k 元组的输入列表,生成一个arity元组 k ,其字段是对应的总和列表中元组的字段。

输入

List( (1,2,3), (2,3,-3), (1,1,1)) 

输出

(4, 6, 1)

我试图使用foldLeft,但我无法将其编译。现在,我正在使用 for 循环,但我一直在寻找更简洁的解决方案。

3 个答案:

答案 0 :(得分:7)

使用shapeless

可以安全,简洁地完成此操作
scala> import shapeless._, syntax.std.tuple._
import shapeless._
import syntax.std.tuple._

scala> val l = List((1, 2, 3), (2, 3, -1), (1, 1, 1))
l: List[(Int, Int, Int)] = List((1,2,3), (2,3,-1), (1,1,1))

scala> l.map(_.toList).transpose.map(_.sum)
res0: List[Int] = List(4, 6, 3)

请注意,与依赖强制转换的解决方案不同,此方法是类型安全的,并且在编译时而不是在运行时检测到任何类型错误,

scala> val l = List((1, 2, 3), (2, "foo", -1), (1, 1, 1))
l: List[(Int, Any, Int)] = List((1,2,3), (2,foo,-1), (1,1,1))

scala> l.map(_.toList).transpose.map(_.sum)
<console>:15: error: could not find implicit value for parameter num: Numeric[Any]
              l.map(_.toList).transpose.map(_.sum)
                                          ^

答案 1 :(得分:3)

scala> val tuples = List( (1,2,3), (2,3,-3), (1,1,1))
tuples: List[(Int, Int, Int)] = List((1,2,3), (2,3,-3), (1,1,1))

scala> tuples.map(t => t.productIterator.toList.map(_.asInstanceOf[Int])).transpose.map(_.sum)
res0: List[Int] = List(4, 6, 1)

productIterator上致电Tuple3时,类型信息会丢失,因此您必须从Any转换回Int

如果元组总是包含相同的类型,我建议使用另一个集合,例如ListTuple更适合不同类型。当您使用相同类型并且不使用productIterator丢失类型信息时,解决方案会更优雅。

scala> val tuples = List(List(1,2,3), List(2,3,-3), List(1,1,1))
tuples: List[List[Int]] = List(List(1, 2, 3), List(2, 3, -3), List(1, 1, 1))

scala> tuples.transpose.map(_.sum)
res1: List[Int] = List(4, 6, 1)

答案 2 :(得分:2)

scala> val list = List( (1,2,3), (2,3,-3), (1,1,1))
list: List[(Int, Int, Int)] = List((1,2,3), (2,3,-3), (1,1,1))

scala> list.foldRight( (0, 0, 0) ){ case ((a, b, c), (a1, b1, c1)) => (a + a1, b + b1, c + c1) }
res0: (Int, Int, Int) = (4,6,1)