본문으로 건너뛰기

컬렉션 파트 검색

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 and 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))

}

조건자를 사용하여 가져오거나 삭제할 요소의 수를 정의할 수도 있습니다. 위에 설명된 것과 유사한 네 가지 함수가 있습니다.

  • 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()는 단일 인수(청크 크기)를 사용하고 지정된 크기의 List의 List를 반환합니다. 첫 번째 청크는 첫 번째 요소부터 시작하여 size 요소를 포함하고, 두 번째 청크는 다음 size 요소를 보유합니다. 마지막 청크의 크기가 더 작을 수 있습니다.


fun main() {

    val numbers = (0..13).toList()
    println(numbers.chunked(3))

}

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


fun main() {

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

}

Windowed​

지정된 크기의 컬렉션 요소의 가능한 모든 범위를 검색할 수 있습니다. 해당 범위를 가져오는 함수를 windowed()라고 합니다. 지정된 크기의 슬라이딩 윈도우를 통해 컬렉션을 보는 경우 보게 될 요소 범위 목록을 반환합니다. chunked()와 달리 windowed()는 각 컬렉션 요소부터 시작하는 요소 범위(windows)를 반환합니다. 모든 윈도우는 단일 List의 요소로 반환됩니다.


fun main() {

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

}

windowed()는 선택적 매개변수를 사용하여 더 많은 유연성을 제공합니다.

  • step은 두 개의 인접한 윈도우의 첫 번째 요소 사이의 거리를 정의합니다. 기본적으로 값은 1이므로 결과에는 모든 요소부터 시작하는 윈도우가 포함됩니다. 단계를 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})

}