Learn Python3:函数

128 阅读4分钟
原文链接: www.jianshu.com

一、定义函数

Python中,定义一个函数要使用def语句,依次写出函数名、括号、括号中的参数(如果需要参数)和冒号:,然后在缩进块中(缩进四个空格)编写函数体,有返回值的话用return语句返回。将程序的大部分繁重工作移到函数中去完成,从而简化主程序。

一个简单的函数如下:

def test(name):
    """打印问候语"""
    print('hello ' + name)
    return name

用三引号括起来的文本是文档字符串,Python用它们生成生成函数的文档。

还可以定义空函数:

def test():
    pass

二、函数的参数

1、位置参数

调用函数时,Python要将函数调用中的每个实参关联到函数定义中对应的形参,这种关联方式称为位置参数。

def power(x, y):
    """计算x的y次方"""
    return x**y

x、y就是位置参数,调用power函数时必须传入对应的x、y

>>>power(2, 4)
16
2、默认参数

针对上边的power函数,如果大多数情况只需计算平方,就可以使用默认参数,对power函数做如下修改:

def power(x, y=2):
    return x**y

此时调用power函数时,如果不提供y参数,则默认计算x的平方:

>>>power(3)
9
>>>power(3, 3)
27

注意默认参数需放在必选参数后边,通过默认参数,可以简化函数的调用。

3、关键字参数

关键字参数允许你传入任意个key-value,这些key-value在函数内部自动组装为一个dict。

如果使用关键字参数,不用考虑函数调用中的实参顺序问题:

def person(first_name, last_name):
    return first_name + ' ' + last_name
>>>person(first_name='zhang', last_name='san')
zhang san
>>>person(last_name='san', first_name='zhang')
zhang san

另外,要传递多个key-value如何做呢?

def person(**person_info):
    print(person_info)
# 传入多个key-value
>>>person(name='tom', age=10)
{'name': 'tom', 'age': 10}
# 传入dict
>>>info = {'name': 'tom', 'age': 10}
>>>person(**info)
{'name': 'tom', 'age': 10}

可以发现,我们传入的关键字参数名字没有任何限制,当然可以通过添加条件判断来实现,但是麻烦,命名关键字参数很好的解决了这个问题:

def person(name, *, age, job):
    print(name, age, job)

注意形参列表的星号,星号后的形参就是命名关键字参数,如果传入的关键字参数的名字不在其中,就会报错:

>>>person('tom', age=20, job='it')
tom 20 it
>>>person('tom', age=20, job='it', sex='boy')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: person() got an unexpected keyword argument 'sex'
4、可变参数

顾名思义就是传入的参数个数是可变的,这点和关键字参数类似。

def get_sum(*numbers):
    return sum(numbers)
# 传入多个实参
>>>get_sum(1, 2, 3, 4)
10
# 传入list
>>>l = [1, 2, 3, 4]
>>>get_sum(*l)
10
# 传入tuple
>>>t = (1, 2, 3, 4])
>>>get_sum(*t)
10

三、函数的返回值

函数并非总是直接显示输出,它可以处理一些数据,并返回一个或一组值。 Python的返回值比较特殊,除了返回单个基本类型数据、列表、字典、set外,还可以返回多个值,例如:

def test():
    return 1, 2, 3, 4
>>>test()
(1, 2, 3, 4)

可以看到,Python的函数返回多值其实就是返回一个tuple。

三、函数编写注意事项

  • 应给函数指定描述性名称,且只在其中使用小写字母和下划线。描述性名称可帮助你和别人明白代码想要做什么。
  • 每个函数都应包含简要地阐述其功能的注释,该注释应紧跟在函数定义后面,并采用文档字
    符串格式
  • 给形参指定默认值时,等号两边不要有空格:
def power(x, y=2):
    return x**y

对于函数调用中的关键字实参,也应遵循这种约定:

>>>power(3, y=4)
  • 如果程序或模块包含多个函数,可使用两个空行将相邻的函数分开