컬렉션 일부 가져오기

Kotlin 표준 라이브러리에는 컬렉션의 일부를 가져오기 위한 확장 함수들이 포함되어 있습니다. 이 함수들은 위치를 명시적으로 나열하거나 결과 크기를 지정하는 등, 결과 컬렉션에 포함할 요소를 선택하는 다양한 방법을 제공합니다.

Slice

slice()는 주어진 인덱스에 해당하는 컬렉션 요소의 리스트를 반환합니다. 인덱스는 범위(range) 또는 정수 값의 컬렉션으로 전달할 수 있습니다.

fun main() {
    val numbers = listOf("one", "two", "three", "four", "five", "six")    
    println(numbers.slice(1..3))
    println(numbers.slice(0..4 step 2))
    println(numbers.slice(setOf(3, 5, 0)))    
}

Take와 drop

첫 번째 요소부터 지정된 개수의 요소를 가져오려면 take() 함수를 사용하세요. 마지막 요소들을 가져오려면 takeLast()를 사용합니다. 컬렉션 크기보다 큰 숫자로 호출하면 두 함수 모두 전체 컬렉션을 반환합니다.

처음 또는 마지막의 일정 개수 요소를 제외한 모든 요소를 가져오려면 각각 drop()과 dropLast() 함수를 호출하세요.

fun main() {
    val numbers = listOf("one", "two", "three", "four", "five", "six")
    println(numbers.take(3))
    println(numbers.takeLast(3))
    println(numbers.drop(1))
    println(numbers.dropLast(5))
}

조건식(predicate)을 사용하여 가져오거나 제외할 요소의 개수를 정의할 수도 있습니다. 위에서 설명한 것과 유사한 네 가지 함수가 있습니다:

• takeWhile()은 조건식이 있는 take()입니다. 조건식과 일치하지 않는 첫 번째 요소 이전까지의 요소를 가져옵니다. 첫 번째 컬렉션 요소가 조건식과 일치하지 않으면 결과는 비어 있게 됩니다.
• takeLastWhile()은 takeLast()와 유사합니다. 컬렉션 끝에서부터 조건식과 일치하는 요소 범위를 가져옵니다. 범위의 첫 번째 요소는 조건식과 일치하지 않는 마지막 요소 바로 다음에 있는 요소입니다. 마지막 컬렉션 요소가 조건식과 일치하지 않으면 결과는 비어 있게 됩니다.
• dropWhile()은 동일한 조건식을 사용하는 takeWhile()의 반대입니다. 조건식과 일치하지 않는 첫 번째 요소부터 끝까지의 요소를 반환합니다.
• dropLastWhile()은 동일한 조건식을 사용하는 takeLastWhile()의 반대입니다. 시작부터 조건식과 일치하지 않는 마지막 요소까지의 요소를 반환합니다.

fun main() {
    val numbers = listOf("one", "two", "three", "four", "five", "six")
    println(numbers.takeWhile { !it.startsWith('f') })
    println(numbers.takeLastWhile { it != "three" })
    println(numbers.dropWhile { it.length == 3 })
    println(numbers.dropLastWhile { it.contains('i') })
}

Chunked

컬렉션을 지정된 크기의 부분들로 나누려면 chunked() 함수를 사용하세요. chunked()는 단일 인자인 청크(chunk) 크기를 받으며, 지정된 크기의 List들을 담은 List를 반환합니다. 첫 번째 청크는 첫 번째 요소부터 시작하여 size만큼의 요소를 포함하고, 두 번째 청크는 다음 size만큼의 요소를 포함하는 식으로 이어집니다. 마지막 청크는 크기가 더 작을 수 있습니다.

fun main() {
    val numbers = (0..13).toList()
    println(numbers.chunked(3))
}

반환된 청크에 대해 즉시 변환(transformation)을 적용할 수도 있습니다. 이를 위해 chunked()를 호출할 때 변환을 람다 함수로 제공하세요. 람다 인자는 컬렉션의 청크입니다. 변환과 함께 chunked()가 호출되면, 청크는 해당 람다 내에서 즉시 소비되어야 하는 수명이 짧은(short-living) List입니다.

fun main() {
    val numbers = (0..13).toList() 
    println(numbers.chunked(3) { it.sum() })  // `it`은 원본 컬렉션의 청크입니다
}

Windowed

지정된 크기의 컬렉션 요소들로 구성된 가능한 모든 범위를 가져올 수 있습니다. 이를 위한 함수는 windowed()입니다. 이 함수는 지정된 크기의 슬라이딩 윈도우(sliding window)를 통해 컬렉션을 바라볼 때 볼 수 있는 요소 범위들의 리스트를 반환합니다. chunked()와 달리 windowed()는 각 컬렉션 요소에서 시작하는 요소 범위(windows)를 반환합니다. 모든 윈도우는 단일 List의 요소로 반환됩니다.

fun main() {
    val numbers = listOf("one", "two", "three", "four", "five")    
    println(numbers.windowed(3))
}

windowed()는 기본값이 있는 매개변수를 통해 더 많은 유연성을 제공합니다:

• step은 인접한 두 윈도우의 첫 번째 요소 사이의 거리를 정의합니다. 기본값은 1이므로, 결과에는 모든 요소에서 시작하는 윈도우가 포함됩니다. step을 2로 늘리면 홀수 번째 요소(첫 번째, 세 번째 등)에서 시작하는 윈도우만 받게 됩니다.
• partialWindows는 컬렉션 끝부분의 요소에서 시작하는 더 작은 크기의 윈도우를 포함합니다. 예를 들어, 세 개의 요소로 구성된 윈도우를 요청할 때 마지막 두 요소에 대해서는 이를 구성할 수 없습니다. 이 경우 partialWindows를 활성화하면 크기가 2와 1인 리스트 두 개가 더 포함됩니다.

마지막으로, 반환된 범위에 대해 즉시 변환을 적용할 수 있습니다. 이를 위해 windowed()를 호출할 때 변환을 람다 함수로 제공하세요.

fun main() {
    val numbers = (1..10).toList()
    println(numbers.windowed(3, step = 2, partialWindows = true))
    println(numbers.windowed(3) { it.sum() })
}

두 개의 요소로 구성된 윈도우를 만들기 위해 별도의 함수인 zipWithNext()가 있습니다. 이 함수는 수신 객체 컬렉션의 인접한 요소 쌍을 생성합니다. zipWithNext()는 컬렉션을 쌍으로 쪼개는 것이 아니라, 마지막 요소를 제외한 각 요소에 대해 Pair를 생성하므로, [1, 2, 3, 4]에 대한 결과는 [[1, 2], [2, 3], [3, 4]]가 되며 [[1, 2], [3, 4]]가 아님에 유의하세요. zipWithNext() 역시 변환 함수와 함께 호출할 수 있으며, 이 함수는 수신 객체 컬렉션의 두 요소를 인자로 받아야 합니다.

fun main() {
    val numbers = listOf("one", "two", "three", "four", "five")    
    println(numbers.zipWithNext())
    println(numbers.zipWithNext() { s1, s2 -> s1.length > s2.length})
}