Получение частей коллекции

Фрагмент

slice() возвращает список элементов коллекции с заданными индексами. Индексы можно передать как диапазон или как коллекцию целых значений.

fun main() {
//sampleStart    
    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)))    
//sampleEnd
}

Взятие и пропускание

Для получения указанного количества элементов, начиная с первого, используйте функцию take(). Для получения последних элементов используйте takeLast(). При вызове с числом, большим размера коллекции, обе функции возвращают всю коллекцию.

Для получения всех элементов, кроме заданного количества первых или последних элементов, вызовите соответственно функции drop() и dropLast().

fun main() {
//sampleStart
    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))
//sampleEnd
}

Вы также можете использовать предикаты для определения количества элементов для взятия или пропускания. Существуют четыре функции, аналогичные описанным выше:

• takeWhile() — это take() с предикатом: она берет элементы до, но не включая, первый элемент, не соответствующий предикату. Если первый элемент коллекции не соответствует предикату, результат пуст.

• takeLastWhile() аналогична takeLast(): она берет диапазон элементов, соответствующих предикату, с конца коллекции. Первый элемент диапазона — элемент, следующий за последним элементом, не соответствующим предикату. Если последний элемент коллекции не соответствует предикату, результат пуст;

• dropWhile() — это противоположность takeWhile() с тем же предикатом: она возвращает элементы от первого, не соответствующего предикату, до конца.

• dropLastWhile() — это противоположность takeLastWhile() с тем же предикатом: она возвращает элементы с начала до последнего, не соответствующего предикату.

fun main() {
//sampleStart
    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') })
//sampleEnd
}

Разбиение на части

Чтобы разбить коллекцию на части заданного размера, используйте функцию chunked(). chunked() принимает единственный аргумент — размер фрагмента — и возвращает List из List заданного размера. Первый фрагмент начинается с первого элемента и содержит size элементов, второй фрагмент содержит следующие size элементов и так далее. Последний фрагмент может иметь меньший размер.

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

Вы также можете применить преобразование к возвращаемым фрагментам сразу. Для этого укажите преобразование в виде лямбда-функции при вызове chunked(). Аргументом лямбда-функции является фрагмент коллекции. При вызове chunked() с преобразованием, фрагменты являются кратковременными List и должны быть обработаны непосредственно в этой лямбда-функции.

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

Оконные

Вы можете получить все возможные диапазоны элементов коллекции заданного размера. Функция для их получения называется windowed(): она возвращает список диапазонов элементов, которые вы бы увидели, если бы смотрели на коллекцию через скользящее окно заданного размера. В отличие от chunked(), windowed() возвращает диапазоны элементов (окна), начинающиеся с каждого элемента коллекции. Все окна возвращаются как элементы одного List.

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

windowed() предоставляет большую гибкость с необязательными параметрами:

• step определяет расстояние между начальными элементами двух соседних окон. По умолчанию значение равно 1, поэтому результат содержит окна, начинающиеся со всех элементов. Если вы увеличите шаг до 2, вы получите только окна, начинающиеся с нечётных элементов: первого, третьего и так далее.

• partialWindows включает окна меньшего размера, которые начинаются с элементов в конце коллекции. Например, если вы запрашиваете окна из трёх элементов, вы не можете построить их для последних двух элементов. Включение partialWindows в этом случае включает два дополнительных списка размеров 2 и 1.

Наконец, вы можете применить преобразование к возвращаемым диапазонам сразу. Для этого укажите преобразование в виде лямбда-функции при вызове windowed().

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

Для построения окон из двух элементов существует отдельная функция — zipWithNext(). Она создаёт пары соседних элементов получаемой коллекции. Обратите внимание, что zipWithNext() не разбивает коллекцию на пары; она создаёт Pair для каждого элемента, кроме последнего, поэтому её результат на [1, 2, 3, 4] равен [[1, 2], [2, 3], [3, 4]], а не [[1, 2], [3, 4]]. zipWithNext() также может быть вызвана с функцией преобразования; она должна принимать два элемента получаемой коллекции в качестве аргументов.

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