跳到主要内容

函数

函数​

为了减少重复，函数登场。

定义函数​

在Python中可以使用 def 关键字来定义函数，程序中函数的参数就相当于是数学上说的函数的自变量，可以通过 return 关键字来返回一个值，这相当于数学上说的函数的因变量。

def add(a, b):
    """
    add two nums
    :param a: first num
    :param b: second num
    :return: result
    """
    c = a + b
    return c

使用函数​

使用函数时，只需要将参数换成特定的值传给函数。

# Python并没有限定参数的类型，因此可以使用不同的参数类型：
print(add(2, 3))

print(add('foo', 'bar'))  # foobar

传入参数时，Python提供了两种选项，

第一种是上面使用的按照位置传入参数，

另一种则是使用关键词模式，显式地指定参数的值：

add(a=2, b=3)
add(b='morning', a='good')
add(2, b=3)  # 5

设定默认参数​

def quad(x, a=1, b=0, c=0):
    return a * x * x + b * x + c
quad(2.0)
quad(2.0, b=3)

接收不定参数​

使用如下方法，可以使函数接受不定数目的参数,类似java的..多个参数：

def add(x, *args):
    total = x
    for arg in args:
        total += arg
    return total

*args 表示参数数目不定，可以看成一个元组，

把第一个参数后面的参数当作元组中的元素。

print(add(1, 2, 3, 4, 5))  # 15
print(add(1, 2))  # 3

使用关键词传入参数​

def add(x, **kwargs):
    total = x
    for arg, val in kwargs.items():
        print("adding ", arg)
        total += val
    return total

**kwargs 表示参数数目不定，相当于一个字典，关键词和值对应于键值对。

add(1, a=2, b=3)  # 6
# 可以接收任意数目的位置参数和键值对参数：
def fun1(*args, **kwargs):
    print(args, kwargs)
    
fun1(2, 3, a="bar", b=10)  # (2, 3) {'a': u'bar', 'b': 10}

返回多个值​

# 函数可以返回多个值：
def to_val(x, y):
    r = (x ** 2 + y ** 2) ** 0.5
    total = x + y
    return r, total
a, b = to_val(3, 4)
print(a, b)  # 5.0 7
# 事实上，Python将返回的两个值变成了元组：
print(to_val(3, 4))  # (5.0, 7)
# 列表也有相似的功能,可以用来赋值：
a, b, c = [1, 2, 3]
print(a, b, c)
# 可以将参数用元组传入：
def add(a, b):
    return a + b

c = (2, 3)
print(add(*c))  # 5
# 这里的*必须要。
# 还可以用字典传入参数哦：
d = {'a': 2, 'b': 5}
print(add(**d))  # 7

map 方法生成序列​

map函数

map() 会根据提供的函数对指定序列做映射。

map(aFun, aSeq)

def sqr(x):
    return x ** 2
a = [2, 3, 4]
result = map(sqr, a)  # [4,9,16]
type(result)
# map返回的是个迭代器对象, 可以转化为list显示

list(result)

事实上，根据函数参数的多少，map 可以接受多组序列， 将其对应的元素作为参数传入函数：

def add(a, b):
    return a + b

a = [2, 3, 4]
list(map(sqr, a))  # [4,9,16]
a = (2, 3, 4)
b = [10, 11, 15]
list(map(add, a, b))  # [12, 14, 19]

reduce​

reduce() 函数会对参数序列中元素进行累积。

from functools import reduce

def add(x, y) :            # 两数相加
    return x + y
sum1 = reduce(add, [1,2,3,4,5])   # 计算列表和：1+2+3+4+5
sum2 = reduce(lambda x, y: x+y, [1,2,3,4,5])  # 使用 lambda 匿名函数
print(sum1)
print(sum2)