컬렉션 일부 가져오기

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

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

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

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

fun main() {
    val numbers = (0..13).toList() 
    println(numbers.chunked(3) { it.sum() })  // `it` is a chunk of the original collection
}

Windowed

주어진 크기의 컬렉션 요소에 대한 모든 가능한 범위를 검색할 수 있습니다. 이를 가져오는 함수는 windowed()라고 불립니다. 이 함수는 주어진 크기의 슬라이딩 윈도우(sliding window)를 통해 컬렉션을 볼 때 나타나는 요소 범위(윈도우)의 리스트를 반환합니다. chunked()와 달리 windowed()는 각 컬렉션 요소부터 시작하는 요소 범위(윈도우)를 반환합니다. 모든 윈도우는 단일 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()가 있습니다. 이 함수는 리시버(receiver) 컬렉션의 인접한 요소 쌍을 생성합니다. 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})
}