在Scala中获取子数组的正确方法是什么?


94

我正在尝试在scala中获得一个子数组,而我对执行此操作的正确方式有些困惑。我最想要的是类似如何在python中完成的工作:

x = [3, 2, 1]
x[0:2]

但我相当确定您不能这样做。

最明显的方法是使用Java Arrays util库。

import java.util.Arrays
val start = Array(1, 2, 3)
Arrays.copyOfRange(start, 0, 2)

但这总是让我觉得在Scala中使用Java库有点脏。我发现最“ scalaic”的方式是

def main(args: List[String]) {
    val start = Array(1, 2, 3)
    arrayCopy(start, 0, 2)
}
def arrayCopy[A](arr: Array[A], start: Int, end: Int)(implicit manifest: Manifest[A]): Array[A] = {
    val ret = new Array(end - start)
    Array.copy(arr, start, ret, 0, end - start)
    ret
}

但是有更好的方法吗?

Answers:



13

假设您有一个数组,其中包含从a到的元素f

scala> val array = ('a' to 'f').toArray // Array('a','b','c','d','e','f')

然后,您可以通过不同的方式从中提取一个子数组:

  1. 删除前n个第一个元素 drop(n: Int)

    array.drop(2) // Array('c','d','e','f')

  2. 取前n个元素 take(n: Int)

    array.take(4) // Array('a','b','c','d')

  3. 使用选择任意元素间隔slice(from: Int, until: Int)。请注意,这until是排除在外的。

    array.slice(2,4) // Array('c','d')

    slice方法严格等效于:
    array.take(4).drop(2) // Array('c','d')

  4. 用排除最后n个元素dropRight(n: Int)

    array.dropRight(4) // Array('a','b')

  5. 使用以下命令选择最后n个元素takeRight(n: Int)

    array.takeRight(4) // Array('c','d','e','f')

参考官方文档


0

从2D Scala数组(original_array)提取特定列的示例:

import scala.collection.mutable.ArrayBuffer

val sub_array = ArrayBuffer[Array[String]]()
val columns_subset: Seq[String] = Seq("ColumnA", "ColumnB", "ColumnC")
val columns_original = original_array(0)

for (column_now <- columns_subset) {
      sub_array += original_array.map{_(columns_original.indexOf(column_now))}
    }
sub_array
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.