Spark记录-Scala函数
Scala函数
Scala有函数和方法。
Scala方法是一个具有名称和签名的类的一部分。 Scala中的函数是一个可以分配给变量的完整对象。
函数定义可以出现在源文件中的任何位置。
不带参数的函数
在Scala中定义函数,请使用def
关键字,后跟方法名和方法体,如下所示。
def hello() = {"Hello World!"}
等号=
用作方法签名和方法体之间的分隔符。
我们可以使用hello()
或hello
调用此函数。
object Main {
def main(args: Array[String]) {
def hello() = {"Hello World!"}
println(hello );
}
}
注意
我们还可以包括可选的返回类型,如下所示。
def hello():String = {"Hello World!"}
我们可以从方法体中完全删除括号。
def hello() = "Hello World!"
我们也可以从方法签名中删除括号。
def hello = "Hello World!"
带参数的函数
以下代码显示如何使用参数创建函数。
def square (i:Int) = {i*i}
函数的主体是表达式, 其中最后一行变为函数的返回值。
我们可以调用这个函数为square(2)。
object Main { def main(args: Array[String]) { def square (i:Int) = {i*i} println(square(2) ); } }
我们可以在函数中提供多个参数。多个参数之间用逗号分隔,如以下示例所示。
def add(x: Int, y: Int): Int = { x + y }
我们现在可以通过将实际参数传递给add函数来调用这个函数。
object Main { def main(args: Array[String]) { def add(x: Int, y: Int): Int = { x + y } println(add(5, 5) ); } }
Scala函数编程
Scala允许我们将函数表达为文字。
函数文字允许我们有一个函数类型的表达式,我们可以写一个短格式,而不声明它的名称。
函数类型可以是以下之一:
- 可以为其分配函数的变量或参数的类型
- 采用函数参数的高阶函数的参数
- 返回函数的高阶函数的结果类型
函数常量以括号括起来的逗号分隔的参数列表开头,后跟箭头和函数体。
函数常量也称为匿名函数。
考虑一个add函数:
val add = (x: Int, y: Int) => x + y
使用函数常量,您可以定义添加函数,如下所示:
(x: Int, y: Int) => x + y.
函数常量被实例化为称为函数值的对象。
函数对象扩展FunctionN traits中的一个,例如Function0,Function1等等直到Function22。
根据函数中的参数数量,相应的FunctionN trait由编译器选择。
对于具有两个参数的函数,编译器选择Function2作为底层类型。对于具有3个参数的函数,编译器选择Function3,对于具有4个参数的函数,Function4等。
因为函数值是一个对象,它可以存储在一个变量中,它可以使用括号函数调用来调用,如下所示:
object Main extends App {
val add = (a: Int, b: Int) => a + b
println(add(1, 2));
}
一级函数和高阶函数
Scala函数是对象。
在函数式编程中,函数是一等公民函数。 一等公民函数可以
- 分配给变量,
- 作为参数传递给另一个函数
- 作为其他函数的值返回。
将函数作为参数或返回函数的函数称为高阶函数。
函数作为变量
正如我们可以传递String,Int和其他变量一样,我们可以像一个变量一样传递一个函数。
我们可以定义一个函数常量,然后将该常量赋值给一个变量。
以下代码定义了一个函数常量,它接受一个Int参数,并返回一个值,该值是传递的Int的两倍:
(i: Int) => { i * 2 }
我们现在可以将该函数常量分配给一个变量:
val doubler = (i: Int) => { i * 2 }
变量doubler
是一个函数的实例,称为函数值。
我们现在可以调用doubler,如下所示:
doubler(2)
doubler
是Function1 trait的一个实例,它定义了一个接受一个参数的函数。
doubler
是使用关键字val
创建并分配给变量的函数。
函数作为参数
我们可以创建一个函数或一个方法,它将一个函数作为参数。
为此,首先定义一个将函数作为参数的方法。
def operation(functionparam:(Int, Int) => Int) { println(functionparam(4,4)) }
操作方法使用一个名为functionparam的参数,它是一个函数。
functionparam函数接受两个Int并返回一个Int
。
操作方法返回一个Unit,指示操作方法不返回任何内容。
接下来,定义一个与预期签名匹配的函数。以下add函数匹配该签名,因为它需要两个Int参数并返回Int:
val add = (x: Int, y:Int) => { x + y }
现在我们可以将一个add函数传递给操作方法:
object Main extends App {
def operation(functionparam:(Int, Int) => Int) {
println(functionparam(4,4))
}
val add = (x: Int, y:Int) => { x + y }
operation(add)
}
任何匹配此签名的函数都可以传递到操作方法中。
object Main extends App {
def operation(functionparam:(Int, Int) => Int) {
println(functionparam(4,4))
}
val add = (x: Int, y:Int) => { x + y }
operation(add)
val subtract = (x: Int, y:Int) => { x - y }
val multiply = (x: Int, y:Int) => { x*y }
operation(subtract)
operation(multiply)
}
返回函数
我们可以从函数或方法返回一个函数。
为了做到这一点,首先定义一个匿名函数。下面的代码声明一个匿名函数,它接受一个String参数并返回一个String:
(name: String) => { "hello" + " " + name }
现在我们将定义一个方法来返回我们刚才定义的匿名函数。
def greeting() = (name: String) => {"hello" + " " + name}
在=符号的左侧有一个正常的方法声明:
def greeting()
在右边的是一个函数文字:
def greeting() = (name: String) => {"hello" + " " + name}
现在你可以将greeting()赋给一个变量:
val greet= greeting()
因为匿名函数接受一个String参数名,我们可以传递一个名字:
object Main extends App {
def greeting() = (name: String) => {"hello" + " " + name}
val greet= greeting()
println(greet("Scala"))
}
附录
object function { def main(args:Array[String]):Unit={ def hello() = {"Hello World!"} //无参数无返回值的函数 def hello1() = "Hello World!" def hello2 = "Hello World!" println(hello ); def hello3():String = {"Hello World!"} //无参数返回值为String的函数 def square (i:Int) = {i*i} //带参数无返回值的函数 println(square(2) ); def add(x: Int, y: Int): Int = { x + y } //带多个逗号隔开参数且返回值为Int的函数 println(add(3,4)) val add1 = (x: Int, y: Int) => x + y //函数作为变量 def operation(functionparam:(Int, Int) => Int) { //函数作为参数 println(functionparam(4,4)) } println(operation(add1)) val subtract = (x: Int, y:Int) => { x - y } val multiply = (x: Int, y:Int) => { x*y } operation(subtract) operation(multiply) //返回函数 def greeting() = (name: String) => {"hello" + " " + name} val greet= greeting() println(greet("Scala")) } }