scala的高阶函数

58 阅读2分钟

什么是高阶函数

一种特殊的函数,它的参数或者是返回值是函数

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

map函数

map:映射 把一个元素改成另一个元素,它的参数是一个函数

语法: 新数组 = 旧数组.map(旧元素=>新元素)

案例:把arr中每个元素都变成自己的2倍,并返回一个新数组

def main(args: Array[String]): Unit = {
  // 定义一个数组
  val arr = Array(1,2,3,4)  // Array表示数组,map就是它的方法
 
  val arr1 = arr.map(x => x * 2) 

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

练习1:把(1,2,3,4,5) → (1,4,9,16,25)

val arr = Array(1,2,3,4,5)
val arr1 = arr.map(x => x * x)

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

练习2:把(-1,2,-3,-4)→(1,2,3,4)

val arr = Array(-1,-2,-3,-4)
val arr1 = arr.map(x => if(x >= 0) x else -x)  //x的绝对值

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

foreach函数

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

def main(args: Array[String]): Unit = {
    // 定义一个数组
    val arr = Array(-1,2,-3,-4,5)
    
  // 依次打印每个元素
    arr.foreach(println)

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

  }

filter函数

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

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

案例:只保留数组中的偶数,并存放在一个新的数组中

def main(args: Array[String]): Unit = {
 
  val arr = Array(1,2,3,4,5)
  var arr1 = arr.filter(x => x%2 == 0)

  arr1.foreach(println)
  
}

练习1:去掉不及格的分数:(60,59,80,70) → (60,80,70)

def main(args: Array[String]): Unit = {
  val arr = Array(60,59,80,70)
  var arr1 = arr.filter(x => x >= 60)

  arr1.foreach(println)

}

练习2:对成绩(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)
  var arr1 = arr.map(x => if( x >= 56 && x <= 59) 60 else x)

  arr.foreach(println)

}