一. 什么是高阶函数
高阶函数: 是一种特殊的函数,特殊之处在于:它指使用其他函数作为参数或者返回值。
Scala的常见高阶函数包括Array.map、flatten、flatmap、foreach、reduce、filter和fold等。
二. map函数
任务: 把数组(1,2,3,4,5) → (2,4,6,8,10)
作用: 通过对数组的每一个元素应用一个函数来构建一个新数组。
语法: 新数组 = 旧数组.map(旧元素=>新元素)
def main(args: Array[String]): Unit = {
val arr = Array(1,2,3,4,5)
val fn = (x:Int) => x * 2 // fn 是一个字面量函数
val arr1 = arr.map( fn ) // map是一个高阶函数,因为它的参数类型是函数
// 用for循环输出
for(i <- arr1) {
println(i)
}
}
简化代码:
def main(args: Array[String]): Unit = {
val arr = Array(1,2,3,4,5)
// 没有必要去单独定义一个函数fn,这里直接写在map函数内,此时,它就是一个匿名函数
val arr1 = arr.map(x => x * 2)
// 用for循环输出
for(i <- arr1) {
println(i)
}
}
结果如图:
练习1: 把(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 arr1 = arr.map(x => x * x)
// 用for循环输出
for(i <- arr1) {
println(i)
}
结果如图:
练习2: 把(-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循环输出
for(i <- arr1) {
println(i)
}
结果如图:
三. foreach函数
任务: 打印数组中的元素
作用: 对数组的每一个元素应用函数。
语法: 数组.foreach(元素=>{})
/**
* 数组
* 1、把数组中每个元素都打印出来
* foreach:依次使用当前的函数来处理数组中的每一个元素
*/
def main(args: Array[String]): Unit = {
val arr = Array(-1,2,-3,-4,5)
// arr.foreach(x => { println(x) })
arr.foreach(println)
}
结果如图:
四. filter函数
任务: 在数组中找出元素是偶数的,并组成新数组。(1,2,3,4,5) → (2,4)
作用: 对数组元素使用函数进行过滤。
语法: 新数组= 旧数组.filter(元素=>bool) 它的参数是一个函数,这函数的返回值是布尔值。如果返回为true就保留。
/**
* 数组
* 1、把数组中的偶数保留下来,组成新数组
* filter : 过滤
* filter(函数fn),如果fn的返回值是true,就保留这个元素
*/
def main(args: Array[String]): Unit = {
val arr = Array(1,2,3,4,5)
// xx%2 ==0 用来判断是否为偶数
val arr1 = arr.filter(x => x%2 == 0)
// 打印新数组
arr1.foreach(println)
}
结果如图:
练习: 去掉不及格的分数:(60,59,80,70) → (60,80,70)
def main(args: Array[String]): Unit = {
var arr = Array(60,59,80,70).filter(x => x >= 60)
arr.foreach(println)
}
结果如图:
五. flatten函数
作用: 将二维数组中的元素展平至单数组中。
语法: 二维数组.flatten。不需要写()
/**
* 二维数组
* flatten 不要加()的函数!
*/
def main(args: Array[String]): Unit = {
var arr1 = Array( Array(1,2,3), Array(4,5,6) );
// arr1(0).foreach(println)
println(arr1(0)(0))
val arr2 = arr1.flatten
arr2.foreach(println)
}
结果如图: