Scala | 高阶函数<1>

高阶函数

高阶函数：是一种特殊的函数

特殊之处在于: 它指使用其他函数作为参数或者返回值。

Scala的常见高阶函数包括 Array.map、flatten、flatmap、foreach、reduce、filter和fold等。

def main(args: Array[String]): Unit = {
  // 定义一个数组
  val arr = Array(1,2,3,4)
  // 提问：把arr中每个元素都变成自己的2倍 并返回一个新数组？

  val fn = (x:Int) => x * 2

  //Array表示数组 map就是他的方法
  //map 就是一个高阶函数 他的参数是一个函数

  val arr1 = arr.map(fn)

  for(i <- arr1) {
    println(i)
  }
}

一、map 函数

map: 映射 把一个元素 改成 另一个元素

案例一： 把 (1,2,3,4,5) ===> (1,4,9,16,25)

def main(args: Array[String]): Unit = {
  val arr = Array(1,2,3,4,5)

  val fn = (x:Int) => x * x

  val arr1 = arr.map(fn)

  for(i <- arr1) {
    println(i)
  }
}

案例二： 把 (-1,2,-3,-4) ===> (1,2,3,4)

def main(args: Array[String]): Unit = {
  val arr = Array(-1, 2, -3, -4)

  val arr1 = arr.map(x => if (x >= 0) x else -x)

  for (i <- arr1) {
    println(i)
  }
  
}

二、foreach 函数

作用：对数组的每一个元素应用函数。

语法：数组.foreach(元素=>{})

def main(args: Array[String]): Unit = {
    // 定义一个数组
    val arr = Array(-1,2,-3,-4,5)

//    arr.foreach(x =>{
//      print(x)
//    })
    // 依次打印每个元素
    arr.foreach(println)

    // 依次打印每个元素
//    for(i <- arr) {
//      println(i)
//    }
  }

三、filter 函数

作用：对数组元素使用函数进行过滤。

语法：新数组= 旧数组.filter(元素=>bool)

它的参数是一个函数，这函数的返回值是布尔值。如果返回为true就保留。

def main(args: Array[String]): Unit = {
  // 定义一个数组
  val arr = Array(1,2,3,4,5)

  // 只保留arr中的偶数 并存放在一个新的数组中
  // filter 的参数是一个函数
  // 如果这个函数返回true 则当前的元素会保留下来
  var arr1 = arr.filter(x => x%2 == 0)

  arr1.foreach(println)
}

案例三： 去掉不及格的分数：(60,58,70,88,45,97) → (60,70,88,97)

def main(args: Array[String]): Unit = {
  // 定义一个数组
  val arr = Array(60,58,70,88,45,97)

  // 只保留arr中的偶数 并存放在一个新的数组中
  // filter 的参数是一个函数
  // 如果这个函数返回true 则当前的元素会保留下来
  var arr1 = arr.filter(x => x >= 60)

  arr1.foreach(println)
}

案例四： 对成绩(100,51,55,67,59,89)进行处理，如果分数在56-59之间，就设置为60分，得到的新成绩为(100,51,55,67,60,89)

//对成绩 (100,51,55,67,59,89) 进行处理
  // 如果分数在56-59之间，就设置为60分，得到的新成绩为(100,51,55,67,60,89)
  def main(args: Array[String]): Unit = {
  
    val arr = Array(100,51,55,67,59,89)

    val arr1 = arr.map(x => if(x >= 56 && x <= 59) 60 else x)

//    for(i <- arr1) {
//      println(i)
//    }
    arr1.foreach(println)
  }