[Kotlin] Stream 함수 #2.필터링 함수

📎 앞 포스팅에 이어지는 시리즈(?)

[Kotlin] Stream 함수 #1.Map이라는 단어가 들어가는 함수

 

---

1️⃣ first() / last()

• first() : 컬렉션의 첫 번째 자료를 반환, null 발생 시 오류
• last() : 컬렉션의 마지막 자료를 반환, null 발생 시 오류
• firstOrNull() : 컬렉션의 첫 번째 자료를 반환, null 발생 시 null 반환
• lastOrNull() : 컬렉션의 마지막 자료를 반환, null 발생 시 null 반환

val days = listOf("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday")

fun firstLastExample() {
   println("< first() >")
   println("첫 번째 인자는 " + days.first() + "\n")

   println("< last() >")
   println("마지막 인자는 " + days.last() + "\n")

   println("< 조건이 있는 first() >")
   println("첫 번째 인자는 " + days.first {day -> day.startsWith("W")} + "\n")

   println("< 조건이 있는 last() >")
   println("마지막 인자는 " + days.last {day -> day.startsWith("W")} + "\n")

   println("< firstOrNull() >")
//   println(days.first() {day -> day.startsWith("N")}) -> NoSuchElementException
   println("N으로 시작 하는 요일 : " + days.firstOrNull {day -> day.startsWith("N")} + "\n")

   println("< lastOrNull() >")
//   println(days.last() {day -> day.startsWith("N")}) -> NoSuchElementException
   println("N으로 시작 하는 요일 : " + days.lastOrNull {day -> day.startsWith("N")})
}

/*
< first() >
첫 번째 인자는 Monday

< last() >
마지막 인자는 Sunday

< 조건이 있는 first() >
첫 번째 인자는 Wednesday

< 조건이 있는 last() >
마지막 인자는 Wednesday

< firstOrNull() >
N으로 시작 하는 요일 : null

< lastOrNull() >
N으로 시작 하는 요일 : null
*/

---

2️⃣ filter()

• 컬렉션 내의 조건에 일치하는 자료 반환

val days = listOf("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday")

fun filterExample() {
   days.filter { day -> day.contains("S") }.forEach { print("$it ") }
}

/*
Saturday Sunday
*/

---

3️⃣ take() / takeLast()

• take() : 앞에서부터 인자로 받은 개수 만큼 인자로 갖는 List 반환
• takeLast() : 뒤에서부터 인자로 받은 개수 만큼 인자로 갖는 List 반환
• takeWhile() : 앞에서부터 조건이 false가 될 때까지 조건에 부합하는 인자들을 갖는 List 반환
• takeLastWhile() : 뒤에서부터 조건이 false가 될 때까지 조건에 부합하는 인자들을 갖는 List 반환

val days = listOf("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday")

fun takeExample() {
   println("< take() >")
   days.take(2).forEach { print("$it ") }
   println("\n")

   println("< takeLast() >")
   days.takeLast(2).forEach { print("$it ") }
   println("\n")

   println("< takeWhile() >")
   days.takeWhile { day -> day.length == 6 }.forEach { print("$it ") }
   println("\n")

   println("< takeLastWhile() >")
   days.takeLastWhile { day -> day.startsWith("S") }.forEach { print("$it ") }
}

/*
< take() >
Monday Tuesday 

< takeLast() >
Saturday Sunday 

< takeWhile() >
Monday 

< takeLastWhile() >
Saturday Sunday 
*/

---

4️⃣ drop() / dropLast()

• take 함수와 반대
• drop() : 앞에서부터 인자로 받은 개수 만큼 제외한 List 반환
• dropLast() : 뒤에서부터 인자로 받은 개수 만큼 제외한 List 반환
• dropWhile() : 앞에서부터 조건이 false가 될 때까지 조건을 만족하는 인자를 제외한 List 반환
• dropLastWhile() : 뒤에서부터 조건이 false가 될 때까지 조건을 만족하는 인자를 제외한 List 반환

val days = listOf("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday")

fun dropExample() {
   println("< drop() >")
   days.drop(2).forEach { print("$it ") }
   println("\n")

   println("< dropLast() >")
   days.dropLast(2).forEach { print("$it ") }
   println("\n")

   println("< dropWhile() >")
   days.dropWhile { day -> day.length < 9 }.forEach { print("$it ") }
   println("\n")

   println("< dropLastWhile() >")
   days.dropLastWhile { day -> day.length < 9  }.forEach { print("$it ") }
}

/*
< drop() >
Wednesday Thursday Friday Saturday Sunday 

< dropLast() >
Monday Tuesday Wednesday Thursday Friday 

< dropWhile() >
Wednesday Thursday Friday Saturday Sunday 

< dropLastWhile() >
Monday Tuesday Wednesday 
*/

---

5️⃣ distinct()

• distinct() : 중복된 항목 제외 후 반환
• distinctBy() : 특정 조건을 수행했을 때, 중복되는 값 제외 후 반환

fun distinctExample(){
   val numbers = arrayListOf(1, 1, -1, 0, -3, 1, 2, -2, 2, 0)

   println("< distinct() >")
   println(numbers.distinct())
   println()

   println("< distinctBy() >")
   println(numbers.distinctBy { number -> number.absoluteValue }) // 절댓값이 같으면 중복으로 제외
}

/*
< distinct() >
[1, -1, 0, -3, 2, -2]

< distinctBy() >
[1, 0, -3, 2]
*/

---

➕JAVA에서 비슷하게 구현해보기

완전히 똑같이 구현이 안되었음..

특히 뒤에서부터 비교하는 것들은 똑같이 해내지 못했음..

더 공부해서 아예 똑같이 해볼 수 있기를

static String[] days = {"Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"};
static int[] numbers = {1, 1, -1, 0, -3, 1, 2, -2, 2, 0};
private static void filtering() {
   List<String> reverse = new ArrayList<>(List.of(days));
   Collections.reverse(reverse);

   System.out.println("< 퍼스트, 라스트 >");
   System.out.println("첫 번째 인자는 " + Arrays.stream(days).findFirst().orElse(null));
   // System.out.println("N으로 시작 하는 요일 : " + Arrays.stream(days).filter(day -> day.startsWith("N")).findFirst().orElseThrow()); -> NoSuchElementException
   System.out.println("N으로 시작 하는 요일 : " + Arrays.stream(days).filter(day -> day.startsWith("N")).findFirst().orElse(null));
   System.out.println("마지막 인자는 " +  reverse.stream().findFirst().orElse(null));
   // System.out.println(      "N으로 시작 하는 요일 :  " +  reverse.stream().findFirst().orElseThrow());
   System.out.println("N으로 시작 하는 요일 :  " +  reverse.stream().findFirst().orElse(null));

   System.out.println();

   System.out.println("< 필터 >");
   Arrays.stream(days).filter(day -> day.contains("S")).forEach(s -> System.out.print(s + " "));

   System.out.println();
   System.out.println();

   System.out.println("< 테이크 >");
   Arrays.stream(days).takeWhile(day -> day.length() == 6).forEach(s -> System.out.print(s + " "));
   System.out.println();
   reverse.stream().takeWhile(day -> day.startsWith("S")).forEach(s -> System.out.print(s + " ")); // 뒤집어진 채로 출력됨

   System.out.println();
   System.out.println();
   System.out.println("< 드롭 >");
   Arrays.stream(days).dropWhile(day -> day.length() < 9).forEach(s -> System.out.print(s + " "));
   System.out.println();
   reverse.stream().dropWhile(day -> day.length() < 9).forEach(s -> System.out.print(s + " ")); // 뒤집어진 채로 출력됨

   System.out.println();
   System.out.println();
   System.out.println("< 중복 제거 >");
   System.out.println(Arrays.toString(IntStream.of(numbers).distinct().toArray()));
}

/*
< 퍼스트, 라스트 >
첫 번째 인자는 Monday
N으로 시작 하는 요일 : null
마지막 인자는 Sunday
N으로 시작 하는 요일 :  Sunday

< 필터 >
Saturday Sunday 

< 테이크 >
Monday 
Sunday Saturday 

< 드롭 >
Wednesday Thursday Friday Saturday Sunday 
Wednesday Tuesday Monday 

< 중복 제거 >
[1, -1, 0, -3, 2, -2]
*/