python学习笔记(二) - 函数

一. 调用函数

python内置了一些数据类型转换函数，比如int()函数可以把其他数据类型转换为整形

>>> int('123')
123
>>> int(12.34)
12
>>> float('12.34')
12.34
>>> str(1.23)
'1.23'
>>> unicode(100)
u'100'
>>> bool(1)
True
>>> bool('')
False

函数名其实就是只想一个函数对象的引用，完全可以把函数名赋给一个变量，相当于给这个函数起了一个别名。

>>> a = abs # 变量a指向abs函数
>>> a(-1) # 所以也可以通过a调用abs函数
1

二. 定义函数

1. 自定义函数：

定义一个求绝对值的函数my_abs

def my_abs(x):
    if x >= 0:
        return x
    else:
        return -x

2. 空函数：

如果想定义一个什么事都不做的空函数，可以用pass语句：

def nop():
    pass

if age >= 18:
    pass

3. 参数检查：

数据类型检查可以用内置函数isinstance实现：

def my_abs(x):
    if not isinstance(x, (int, float)):   # 只允许整数和浮点数类型的参数
        raise TypeError('bad operand type')
    if x >= 0:
        return x
    else:
        return -x

4. 返回多个值：

函数可以返回多个值吗？答案是肯定的。

import math
def move(x, y, step, angle=0):
    nx = x + step * math.cos(angle)
    ny = y - step * math.sin(angle)
    return nx, ny

x, y = move(100, 100, 60, math.pi / 6)
print x, y    #  151.961524227 70.0

但其实这只是一种假象，python函数返回的仍然是单一值：

r = move(100, 100, 60, math.pi / 6)
print r       # (151.96152422706632, 70.0)

原来返回值是一个tuple！但是，在语法上，返回一个tuple可以省略括号，而多个变量可以同时接受一个tuple，按位置赋给对应的值。

所以，python的函数返回多值其实就是返回一个tuple

三. 函数的参数

python的函数具有非常灵活的参数形态，既可以实现简单的调用，又可以传入非常复杂的参数。

(1) 默认参数

计算任意n次方：n=2为默认参数

当我们调用power(5)时，相当于调用power(5, 2), 而对于n > 2的情况，就必须明确地传入n，如：power(5, 3)

def power(x, n=2):
    s = 1
    while n > 0:
        n = n - 1
        s = s * x
    return s

默认参数可以简化函数的调用，注意：

a. 必选参数在前，默认参数在后

b. 默认参数也可以有多个

c. 当不按顺序提供部分默认参数时，需要把参数名写上(调用时)

(2) 可变参数

可变参数就是传入的参数个数是可变的。允许你传入0个或任意个参数。这些参数在函数调用时自动组装为一个tuple。

给定一组数字a，b，c……，请计算a2 + b2 + c2 + ……。

def calc(*numbers):
    sum = 0
    for n in numbers:
        sum = sum + n * n
    return sum

调用：calc(1, 2, 3) 或 calc(1, 2, 3, 4)， 在函数内部，参数numbers接受到的是一个tuple

如果有一个list或tuple，要调用一个可变参数，可以在list或tuple前面加一个*，把list或tuple的元素变成可变参数传进去

>>> nums = [1, 2, 3]
>>> calc(*nums)
14

可变参数既可以直接传入：func(1, 2, 3)，又可以先组装list或tuple，再通过*args传入：func(*(1, 2, 3))

(3) 关键字参数

关键字参数允许你传入0个或任意个含参数名的参数，这些关键字参数在函数内部自动组装为一个dict。

def person(name, age, **kw):
    print 'name:', name, 'age:', age, 'other:', kw
	
print person('Michael', 30) #name: Michael age: 30 other: {}
print person('Bob', 35, city='Beijing') #name: Bob age: 35 other: {'city': 'Beijing'}
print person('Adam', 45, gender='M', job='Engineer') #name: Adam age: 45 other: {'gender': 'M', 'job': 'Engineer'}
kw = {'city': 'Beijing', 'job': 'Engineer'} 
print person('Jack', 24, **kw) #name: Jack age: 24 other: {'city': 'Beijing', 'job': 'Engineer'}

**kw是关键字参数，kw接受的是一个dict

关键字参数既可以直接传入：func(a=1, b=2)，又可以先组装dict，再通过**kw传入：func(**{'a' : 1, 'b' : 2})

(4) 参数组合

python中可以把必选参数、默认参数、可变参数和关键字参数组合在一起使用。

参数定义的顺序必须是：必选参数、默认参数、可变参数和关键字参数

def func(a, b, c=0, *args, **kw):
    print 'a =', a, 'b =', b, 'c =', c, 'args =', args, 'kw =', kw

func(1, 2) # a = 1 b = 2 c = 0 args = () kw = {}
func(1, 2, c=3) # a = 1 b = 2 c = 3 args = () kw = {}
func(1, 2, 3, 'a', 'b') # a = 1 b = 2 c = 3 args = ('a', 'b') kw = {}
func(1, 2, 3, 'a', 'b', x=99) # a = 1 b = 2 c = 3 args = ('a', 'b') kw = {'x': 99}
args = (1, 2, 3, 4)
kw = {'x': 99}
func(*args, **kw) # a = 1 b = 2 c = 3 args = (4,) kw = {'x': 99}

四. 递归函数

所谓的递归函数就是一个函数在内部调用自身。

比如，计算阶乘：n! = 1 x 2 x 3 x ... x n

def fact(n):
    if n==1:
        return 1
    return n * fact(n - 1)

在计算机中，函数调用时通过栈(stack)这种数据结构实现的，每当进入一个函数调用，栈就会加一层栈帧，每当函数返回，栈就会减一层栈帧。由于栈的大小不是无限的，所以，递归调用的次数过多，会导致栈溢出。

原文参见：http://www.liaoxuefeng.com/