본문 바로가기
Kotlin

[Kotlin] 기초 문법 : flatten() 함수

by Echung 2023. 12. 16.

Flatten 함수


flatten 함수는 다차원 배열 이나 리스트를 1차원 배열로 만들어 주는 함수입니다.

사진 1. flatten 설명

위의 사진을 보면 주어진 배열에서 모든 배열의 요소들을 단일 목록으로 반환해 준다고 합니다. 코드를 보면서 확인해 보겠습니다.

val deepArray = arrayOf(
	arrayOf(1),
    arrayOf(2, 3),
    arrayOf(4, 5, 6)
)

println(deepArray.flatten()) // [1, 2, 3, 4, 5, 6]

deepArray안에 들어있는 요소들을 1차원 배열로 만들어주는 것을 확인할 수 있었습니다. 
Array뿐만 아니라 List에서도 똑같은 결과를 받을 수 있습니다.

val deepList = listOf(
	listOf(1),
    listOf(2, 3),
    listOf(4, 5, 6)
)

println(deepList.flatten()) // [1, 2, 3, 4, 5, 6]

 

Reference


https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/flatten.html

반응형