高阶函数2

98 阅读1分钟

filter函数

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

语法:新数组= 旧数组.filter(元素=>bool)  它的参数是一个函数,这函数的返回值是布尔值。如果返回为true就保留。

/*
filter
   过滤器
   filter(函数)
   如果函数的返回值为true,则元素就保留下来
 */
def main (args: Array[String]): Unit = {
  // 把数值为偶数的留下来
  val arr1 = Array(1,2,3,4,5)
  var arr2 = arr1.filter(x => x%2==0)
  arr2.foreach(println)

  var score = Array(90,98,60,49,100)
  // 把小于80分的值去掉,只保留大于80分的分数
  var score1 = score.filter(x => x>80).foreach(println)
}

flatten函数

作用:将二维数组中的元素展平至单数组中。

语法:二维数组.flatten。不需要写()

var arr1 = Array(Array(1,2,3), Array(4,5,6))  
println(arr1(0)(0))  
var arr2 = arr1.flatten