6. 函数

用 def 定义一个新函数

def add(x, y):
    print("x is {} and y is {}".format(x, y))
    # 用 return 语句返回
    return x + y

调用函数

# 打印 "x is 5 and y is 6" 并且返回 11
add(5, 6)

也可以用关键字参数来调用函数

# 关键字参数可以用任何顺序
add(y=6, x=5)

还可以定义一个可变参数函数

def varargs(*args):
    return args

# => (1, 2, 3)
varargs(1, 2, 3)

也可以定义一个关键字可变参数函数

def keyword_args(**kwargs):
    return kwargs

来看看结果是什么：

# => {"big": "foot", "loch": "ness"}
keyword_args(big="foot", loch="ness")

这两种可变参数可以混着用

"""
all_the_args(1, 2, a=3, b=4) prints:
    (1, 2)
    {"a": 3, "b": 4}
"""
def all_the_args(*args, **kwargs):
    print(args)
    print(kwargs)

调用可变参数函数时可以做跟上面相反的，用 * 展开元组，用 ** 展开字典。

args = (1, 2, 3, 4)
kwargs = {"a": 3, "b": 4}
# 相当于 all_the_args(1, 2, 3, 4)
all_the_args(*args)
# 相当于 all_the_args(a=3, b=4)
all_the_args(**kwargs)
# 相当于 all_the_args(1, 2, 3, 4, a=3, b=4)
all_the_args(*args, **kwargs)

使用返回多个数值（返回值为元组类型）

def swap(x, y):
    # 用不带括号的元组的格式来返回多个数值
    # (注意: 括号不需要加，但是也可以加)
    return y, x

x = 1
y = 2
# x = 2, y = 1
x, y = swap(x, y)
# 同上，括号不需要加，但是也可以加
(x, y) = swap(x,y)

函数的作用域

x = 5

def setX(num):
    # 局部作用域的 x 和全局域的 x 是不同的
    # => 43
    x = num
    # => 43
    print (x)

def setGlobalX(num):
    global x
    # => 5
    print (x)
    # 现在全局域的 x 被赋值
    x = num
    # => 6
    print (x)

setX(43)
setGlobalX(6)

函数在 Python 中是一等公民

def create_adder(x):
    def adder(y):
        return x + y
    return adder

add_10 = create_adder(10)
# => 13
add_10(3)

也有匿名函数

# => True
(lambda x: x > 2)(3)
# => 5
(lambda x, y: x ** 2 + y ** 2)(2, 1)

内置的高阶函数

# => [11, 12, 13]
list(map(add_10, [1, 2, 3]))
# => [4, 2, 3]
list(map(max, [1, 2, 3], [4, 2, 1]))
# => [6, 7]
list(filter(lambda x: x > 5, [3, 4, 5, 6, 7]))

用列表推导式可以简化映射和过滤，列表推导式的返回值是另一个列表。

# => [11, 12, 13]
[add_10(i) for i in [1, 2, 3]]
# => [6, 7]
[x for x in [3, 4, 5, 6, 7] if x > 5]

也可以用这种方式实现对集合和字典的构建

# => {'d', 'e', 'f'}
{x for x in 'abcddeef' if x not in 'abc'}
# => {0: 0, 1: 1, 2: 4, 3: 9, 4: 16}
{x: x**2 for x in range(5)}