偏函数(partial function)
偏函数(Partial Function),是一个数学概念它不是"函数"的一种, 它跟函数是平行的概念。
Scala中的Partia Function是一个Trait,其的类型为PartialFunction[A,B],其中接收一个类型为A的参数,返回一个类型为B的结果。
基本介绍
- 在对符合某个条件,而不是所有情况进行逻辑操作时,使用偏函数是一 个不错的选择
- 将包在大括号内的一组case语句封装为函数,我们称之为偏函数,它只 对会作用于指定类型的参数或指定范围值的参数实施计算,超出范围的 值会忽略(未必会忽略,这取决于你打算怎样处理)
- 偏函数在Scala中是一个特质PartialFunction
给你一个集合val list = List(1,2,3,4,"abc),请完成如下要求:
- 将集合 list 中的所有数字 +1,并返回一个新的集合
- 要求忽略掉 非数字 的元素,即返回的新的集合形式为 (2,3,4,5)
解决方式-map返回新的集合, 引出偏函数
思路1-map+filter 方式
val list = List(1, 2, 3, 4, "abc")
//思路1,使用map+fliter的思路
def f1(n:Any): Boolean = {
n.isInstanceOf[Int]
}
def f2(n:Int): Int = {
n + 1
}
def f3(n:Any): Int ={
n.asInstanceOf[Int]
}
val list2 = list.filter(f1).map(f3).map(f2)
println("list2=" + list2)
思路2-模式匹配
def addOne2( i : Any ): Any = {
i match {
case x:Int => x + 1
case _ =>
}
}
val list = List(1, 2, 3, 4, "abc")
val list2 = list.map(addOne2)
println("list2=" + list2
使用偏函数解决前面的问题
val list = List(1, 2, 3, 4, "abc")
//说明
val addOne3= new PartialFunction[Any, Int] {
def isDefinedAt(any: Any) = if (any.isInstanceOf[Int]) true else false
def apply(any: Any) = any.asInstanceOf[Int] + 1
}
val list3 = list.collect(addOne3)
println("list3=" + list3) //?
偏函数小结
- 使用构建特质的实现类(使用的方式是
PartialFunction
的匿名子类) PartialFunction
是个特质(看源码)- 构建偏函数时,参数形式
[Any, Int]
是泛型,第一个表示参数类型,第二个表示 返回参数 - 当使用偏函数时,会遍历集合的所有元素,编译器执行流程时先执行
isDefinedAt()
如果为true
,就会执行apply
, 构建一个新的Int
对象返回 - 执行
isDefinedAt()
为false
就过滤掉这个元素,即不构建新的Int对象. map
函数不支持偏函数,因为map
底层的机制就是所有循环遍历,无法过滤处理 原来集合的元素collect
函数支持偏函数
偏函数简化形式
-
简化形式1
def f2: PartialFunction[Any, Int] = { case i: Int => i + 1 // case语句可以自动转换为偏函数 } val list2 = List(1, 2, 3, 4,"ABC").collect(f2)
-
简化形式2
val list3 = List(1, 2, 3, 4,"ABC").collect{ case i: Int => i + 1 } println(list3)