[코틀린] 컬렉션 함수

컬렉션을 반복문을 이용하여 포함된 요소들을 하나씩 꺼내서 사용했다.

하지만 코틀린은 함수형 언어의 특징을 가지고 있기 때문에 좀 더 편리하게 컬렉션을 사용할 수 있다.

 

컬렉션 함수

• list, set, map과 같은 컬렉션 또는 배열에 일반 함수 또는 람다 함수 형태를 사용하여 반복문 없이도 요소를 참조하여 순회하거나 조건을 걸고 구조의 변경까지 가능한 여러가지 함수를 지칭한다.

 

forEach

컬렉션 객체에 사용하면 중괄호 안에서 컬렉션에 포함된 모든 요소를 it이라는 변수로 순서대로 참조할 수 있다.

collection.forEach{
	println(it)
}

 

 

filter

컬렉션 객체에 사용하면 중괄호 안에서 it에 조건을 걸어 조건에 맞는 객체만 다시 컬렉션으로 만들어 반환한다.

collection.forEach{
	it < 4
}

 

map

중괄호 안에서 it에 수식을 적용하여 일괄적으로 요소들의 값을 변경한 후 변경후의 객체를 다시 컬렉션으로 만들어 반환한다.

collection.forEach{
	it * 4
}

 

 

any, all, none

• 중괄호 안에서 it에 조건을 걸어준 뒤
  • any : 하나라도 조건에 맞으면 true를 반환한다.
  • all : 모두 조건에 맞으면 true를 반환한다.
  • none : 하나라도 조건에 맞지 않으면 true를 반환한다.

 

first

일반함수로 사용하는 경우 첫번째 요소를 반환한다.

람다함수 형태로 사용하는 경우 중괄호 안에 it의 조건을 걸어주면, 조건에 맞는 첫번째 요소를 반환한다.

find 함수로 대체할 수 있다.

 

last

일반함수로 사용하는 경우 마지막 요소를 반환한다.

람다함수 형태로 사용하는 경우 중괄호 안에 it의 조건을 걸어주면, 조건에 맞는 마지막 요소를 반환한다.

findLast 함수로 대체할 수 있다.

 

firstOrNull & lastOrNull

first함수와 last함수는 조건에 맞는 객체가 없는 경우 문제가 발생한다.(NoSuchElementException)

firstOrNull이나 lastOrNull 함수를 사용하면 콜렉션에 객체가 없는경우 Null을 반환한다.

 

count

일반함수로 사용하면 포함된 모든 아이템의 개수를 반환한다.

중괄호와 함께 it에 조건문을 걸면 조건에 맞는 요소의 개수를 반환한다.

 

 

예제

fun main(){
    val nameList = listOf("김혜인", "김철수", "김영희", "홍길동", "이순신")

    nameList.forEach{ print(it + " ") }
    println()

    println(nameList.filter { it.startsWith("김") })

    println(nameList.map{ "이름 : " + it})

    println(nameList.any{ it == "장영실" })
    println(nameList.all{ it.length == 3 })
    println(nameList.none { it.startsWith("박") })

    println(nameList.first { it.startsWith("김")})
    println(nameList.last { it.startsWith("김")})
    println(nameList.count { it.startsWith("홍")})
}

김혜인 김철수 김영희 홍길동 이순신 
[김혜인, 김철수, 김영희]
[이름 : 김혜인, 이름 : 김철수, 이름 : 김영희, 이름 : 홍길동, 이름 : 이순신]
false
true
true
김혜인
김영희
1

 

accociateBy

• 컬렉션의 객체(요소)에서 key를 추출하여 map으로 변환하는 함수다.

collection.associateBy{it.name}

 

groupBy

• key가 같은 요소끼리 배열로 묶어 map으로 만드는 함수다.

collection.groupBy{it.birthYear}

 

partition

• 아이템에 조건을 걸어 두개의 컬렉션으로 나누어 준다.
• 나누어진 두 컬렉션은 Pair라는 클래스에 담겨 하나의 객체로 반환된다.
• Pair에서 두 컬렉션은 first, second로 참조한다.
• 또는 변수 선언시 괄호로 변수를 두 개 선언해주면 각각의 변수 이름으로 받을 수 있다.

collection.partition{it.birthYear > 2022}
//또는
val(over2022, under2022) = collection.partition{it.birthYear > 2022}

 

예제

fun main(){
    data class Person(val name: String, val birthYear: Int)

        val personList = listOf(Person("혜인", 1000),
                Person("철수", 2000),
                Person("영희,", 2001),
                Person("철수", 2004))

        println(personList.associateBy { it.birthYear })
        println(personList.groupBy { it.name })

        val(over2002, under2002) = personList.partition { it.birthYear > 2002 }
        println(over2002)
        println(under2002)
}

{1000=Person(name=혜인, birthYear=1000), 2000=Person(name=철수, birthYear=2000), 2001=Person(name=영희,, birthYear=2001), 2004=Person(name=철수, birthYear=2004)}
{혜인=[Person(name=혜인, birthYear=1000)], 철수=[Person(name=철수, birthYear=2000), Person(name=철수, birthYear=2004)], 영희,=[Person(name=영희,, birthYear=2001)]}
[Person(name=철수, birthYear=2004)]
[Person(name=혜인, birthYear=1000), Person(name=철수, birthYear=2000), Person(name=영희,, birthYear=2001)]

 

flatMap

• 요소마다 만들어진 컬렉션을 합쳐서 반환한다.
• 중괄호에서 요소마다 새로운 컬렉션을 생성하면 이를 합쳐서 하나의 컬렉션으로 반환한다.
• collection.flatMap{ listOf(it*3, it+3) }

컬렉션에 2, 3 이 들어있었다면 반환된 컬렉션은 6, 5, 9, 6 이다.

 

getOrElse

• 괄호안에 인덱스 위치를 넣고, 해당 인덱스 위치에 에 요소가 있으면 요소를 반환하고 없다면 중괄호에 지정한 기본값을 반환한다.

// collection = [6, 5, 9, 6]
collection.getOrElse(1){50}		// 인덱스 1에 위치한 5 반환
collection.getOrElse(8){50}		// 인덱스 8 없으므로 기본값 50 반환

 

zip

• 컬렉션 두 개의 요소를 1대1로 매칭하여 Pair클래스에 담은 후 리스트에 넣어 반환한다.
• 이 때 결과 리스트의 요소 개수는 더 작은 컬렉션을 따라가게 된다.

 

예제

fun main(){
    val numbers = listOf(-3, 7, 2, -10, 1)

        println(numbers.flatMap { listOf(it * 10, it + 10) })
    
        println(numbers.getOrElse(1){50})
        println(numbers.getOrElse(10){50})
    
        val names = listOf("A", "B", "C", "D")
    
        println(names zip numbers)
}

[-30, 7, 70, 17, 20, 12, -100, 0, 10, 11]
7
50
[(A, -3), (B, 7), (C, 2), (D, -10)]

 

 

 

💡컬렉션을 사용해야하는 상황에서 컬렉션 함수를 사용하면 반복문과 조건문을 대부분 대체할 수 있다.

💡컬렉션 함수는 람다 함수를 사용하여 컬렉션을 좀 더 편리하게 조작할 수 있는 장점을 가진 함수다.

💡컬렉션 함수는 데이터 조작을 더 수월하게 만들 수 있는 다양한 기능을 가지고 있다.