scala map里面怎么传入函数

Question

scala map里面怎么传入函数

Answer 1

def flatMap1(): Unit = {
val li = List(1,2,3)
val res = li.flatMap(x => x match {
case 3 => List('a','b')
case _ => List(x*2)
})
println(res)
}

def map1(): Unit = {
val li = List(1,2,3)
val res = li.map(x => x match {
case 3 => List('a','b')
case _ => x*2
})
println(res)
}

def main(args: Array[String]): Unit = {
flatMap1()
map1()
}
}