Kotlin collection - Slice / Take and drop / Chunked / Windowed

Posted on | In

Kotlin Collection 공식 문서를 정리 한 글입니다. Collection의 retrieve collection parts operator에 대해 알아봅니다.

Github repo 에서 아래에 적힌 Kotlin 코드들을 확인 하실 수 있습니다.

Retrieve collection parts operator

  • Element들을 가져오는데 쓰이는 operator
  • slice() / take() … 등을 사용

Slice

  • Index를 기반으로 collection을 잘라 새로운 collection을 생성한다.
  • range / collection을 인자로 받을 수 있다.
1
2
3
4
5
6
fun sliceCollectionSampleCode() {
    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

  • collection의 앞 / 뒤에서 부터 n번째 element 까지 잘라 새로운 collection을 생성하는 operator
1
2
3
4
5
6
7
fun takeDropSampleCode() {
    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))
}
  • Boolean 값을 리턴하는 함수와 함께 쓸 수 있음
  • takeWhile() → element의 조건식이 false가 나오기 전까지 element들을 take ( false가 나온 이후의 element들은 생략 )
  • takeLastWhile() → collection의 뒤에서 부터 element의 조건식이 false가 나오기 전까지 element들을 take ( false가 나온 이후의 element들은 생략 )
1
2
3
4
5
6
7
fun takeDropSampleCodeWithPredicates() {
    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

  • collection을 n개씩 잘라 새로운 collection을 만듬
  • n개씩 자르다 남는 마지막 collection의 크기는 n보다 작거나 같음
1
2
3
4
fun chunkSampleCode() {
    val numbers = (0..13).toList()
    println(numbers.chunked(3))
}
  • chunked된 collection을 transform 할 수 있음
1
2
3
4
fun chunkWithTransformSampleCode() {
    val numbers = (0..13).toList()
    println(numbers.chunked(3) { it.sum() })  // `it` is a chunk of the original collection
}

Windowed

  • collection을 앞에서 부터 n개 묶어 list를 생성하고, step size ( default size = 1 ) 만큼씩 index를 건너 뛰어 n개씩 묶은 list를 생성 한 뒤 생성된 list를 element로 가지는 collection 생성
  • 예제를 보면 이해가 쉬울 것
1
2
3
4
fun windowedSampleCode() {
    val numbers = listOf("one", "two", "three", "four", "five")
    println(numbers.windowed(3))
}
  • step / partialWindows 파라메터를 활용하여 묶는 조건 추가 가능
  • transform 가능
1
2
3
4
5
6
fun windowedWithParamsSampleCode() {
    val numbers = (1..10).toList()
    println(numbers.windowed(3, step = 2, partialWindows = false))
    println(numbers.windowed(3, step = 2, partialWindows = true))
    println(numbers.windowed(3) { it.sum() })
}
  • zipWithNext() 을 활용하여 list로 반환하는 window와 달리, element를 묶어 Pair객체로 생성하고, transform 함수도 활용 가능
1
2
3
4
5
fun zipWithNextSampleCode() {
    val numbers = listOf("one", "two", "three", "four", "five")
    println(numbers.zipWithNext())
    println(numbers.zipWithNext() { s1, s2 -> s1.length > s2.length})
}

출처 : https://kotlinlang.org/docs/collection-parts.html

"Android" 카테고리의 최근 포스팅

카테고리 모든 글 보기
RxJava 강의 13 - RxJava with Retrofit / Room, 기존의 Callback을 RxStream으로 2021. 10. 17
RxJava 강의 12 - 에러 처리 연산자 ( retry / retrywhen / onErrorResumeNext / onErrorReturn ) 2021. 10. 11
RxJava 강의 11 - 합성 연산자 ( merge / concat / concatEager / zip / combineLatest ) 2021. 09. 30
RxJava 강의 10 - RxBinding / Throttle / Debounce 2021. 09. 29
RxJava 강의 9 - 흐름 제어 연산자 ( take / filter / map / flatMap / switchMap ... ) 2021. 09. 26
RxJava 강의 8 - 데이터 발행 연산자 ( Create / Just / Defer / FromCallable / Interval ... ) 2021. 09. 25
RxJava 강의 7 - Cold, Hot Observable / Subject ( Processor ) 2021. 09. 15
RxJava 강의 6 - Schedulers 2021. 09. 10
RxJava 강의 5 - Subscribe operator / Consumer, Action interface 2021. 09. 04
RxJava 강의 4 - Observable이란, Stream 구현체의 종류, Disposable 이란 2021. 08. 31