假设有一个List
,例如:
List(1, 2, 3, 4, 5, 6, 7)
什么是将列表拆分为 n 个子列表的最佳方法,并以循环方式将项目放入每个列表中?
例如,如果 n = 3,则结果应为:
List(List(1, 4, 7), List(2, 5), List(3, 6))
我认为在集合API中应该有一种方法可以做到这一点,但我似乎找不到它。
优秀的一行代码能加分哦 ;)
假设有一个List
,例如:
List(1, 2, 3, 4, 5, 6, 7)
什么是将列表拆分为 n 个子列表的最佳方法,并以循环方式将项目放入每个列表中?
例如,如果 n = 3,则结果应为:
List(List(1, 4, 7), List(2, 5), List(3, 6))
scala> def round[T](l: List[T], n: Int) = (0 until n).map{ i => l.drop(i).sliding(1, n).flatten.toList }.toList
round: [T](l: List[T], n: Int)List[List[T]]
scala> round((1 to 7).toList, 3)
res4: List[List[Int]] = List(List(1, 4, 7), List(2, 5), List(3, 6))
这里是一个简单的一行代码:
scala> List.range(1, 10)
res11: List[Int] = List(1, 2, 3, 4, 5, 6, 7, 8, 9)
scala> res11.grouped(3).toList.transpose
res12: List[List[Int]] = List(List(1, 4, 7), List(2, 5, 8), List(3, 6, 9))
很遗憾,当您的列表不可转置时,它将无法工作。
scala> List.range(1, 8).grouped(3).toList.transpose
java.lang.IllegalArgumentException: transpose requires all collections have the
same size
scala> def extend[A](xs: List[A], c: Int): List[Option[A]] = {
| val n = Stream.iterate(c)(c +).find(_ >= xs.length).get
| xs.map(Some(_)).padTo(n, None)
| }
extend: [A](xs: List[A], c: Int)List[Option[A]]
scala> List.range(1, 8)
res13: List[Int] = List(1, 2, 3, 4, 5, 6, 7)
scala> extend(res13, 3).grouped(3).toList.transpose.map(_.flatten)
res14: List[List[Int]] = List(List(1, 4, 7), List(2, 5), List(3, 6))
整合在一起:
scala> def round[A](xs: List[A], c: Int) = {
| val n = Stream.iterate(c)(c +).find(_ >= xs.length).get
| val ys = xs.map(Some(_)).padTo(n, None)
| ys.grouped(c).toList.transpose.map(_.flatten)
| }
round: [A](xs: List[A], c: Int)List[List[A]]
scala> round(List.range(1, 10), 3)
res16: List[List[Int]] = List(List(1, 4, 7), List(2, 5, 8), List(3, 6, 9))
scala> round(List.range(1, 8), 3)
res17: List[List[Int]] = List(List(1, 4, 7), List(2, 5), List(3, 6))
ys.grouped(3)
是一个打字错误吗? - seniaround(list, 0)
会导致 OutOfMemoryError
。 - seniaval list = List(1, 2, 3, 4, 5, 6, 7)
list.grouped(Math.ceil(list.size / 3))