函数
函数是带名字的代码块,用于完成具体工作,通过使用函数,程序的编写、阅读、测试和修复都将更容易
定义函数
使用关键字def
告诉Python要定义一个函数,称为函数定义
,函数名后的括号是必须的,定义以冒号结尾。
在冒号后的所有缩进构成了函数体:
def hello():
print('Hello!')
hello()
要使用这个函数,可调用函数
,可依次指定函数名以及用括号括起来的必要信息。
- 向函数传递信息
在函数定义def hello()
括号内添加username
,可以让函数接受给username
的任何值,调用hello()
时可以将一个名字传递给它
def hello(username):
print("Hello " + username.title() + '!')
hello('jesse')
- 实参和形参
在上面函数hello()
的定义中,变量username
是一个形参——函数完成其工作所需的一项信息。在代码hello('jesse')
中,值'jesse'
是一个实参。实参是调用函数时传递给函数的信息。
在hello('jesse')
中,将实参'jesse'
传递给函数hello()
,这个值被存储在形参username
中。
传递实参
函数调用中可能包含多个实参,想函数传递实参的方式很多,可使用位置实参
,这要求实参的顺序与实参的顺序相同;也可使用关键字实参
,其中每个参数都由变量名和值组成;还可以使用列表和字典。
- 位置实参
调用函数时,Python必须将函数调用中的每个实参都关联到函数定义中的一个形参。为此,最简单的关联方式是基于实参的顺序,称为位置实参
def pet(pet_type, name):
"""显示宠物信息"""
print("I have a " + pet_type + ".")
print("My " + pet_type + "'s name is " + name + ".")
pet('hamster', 'harry')
得到结果:
I have a hamster.
My hamster's name is harry.
这个函数表明,它需要一种动物类型和名字,调用pet()
时,需要按顺序提供类型和名字,实参'hamster'
储存在形参type
中,实参'harry'
存储在形参name
中。
-
调用函数多次
可以根据需要调用函数任意次,调用函数多次是一种效率极高的工作方式。在函数中,可根据需要使用任意数量的为止实参,Python将按顺序将函数调用中的实参关联到函数定义中相应的形参 -
位置实参的顺序很重要
- 关键字实参
关键字实参
是传递给函数的名称-值
对。直接在实参中将名称和值关联起来,因此向函数传递实参时不会混搭。关键字实参无需考虑函数调用中的实参顺序,还清楚地指出了函数调用中各个值的用途。
def pet(pet_type, name):
"""显示宠物信息"""
print("I have a " + pet_type + ".")
print("My " + pet_type + "'s name is " + name + ".")
pet(name='harry', pet_type='hamster')
- 默认值
编写函数时,可给每个形参指定默认值
。在调用函数中给形参提供了实参时,Python将使用指定的实参值,使用默认值可简化函数调用。
def pet(name, pet_type='dog'):
"""显示宠物信息"""
print("I have a " + pet_type + ".")
print("My " + pet_type + "'s name is " + name + ".")
pet('willie')
修改了函数pet()
的定义,在其中给形参pet_type
指定了默认值'dog'
,在待用函数时,如果没有给pet_type
指定值,Python将把这个形参设置为'dog'
需要注意的是,需要将指定了默认值的形参放在后面位置
- 等效的函数调用
鉴于可混合使用位置实参、关键字实参和默认值,通常有多重等效的函数调用方式
def pet(name, pet_type='dog'):
"""显示宠物信息"""
print("I have a " + pet_type + ".")
print("My " + pet_type + "'s name is " + name + ".")
pet('willie')
pet(name='willie')
pet('harry', 'hamster')
pet(pet_type='hamster', name='harry')
pet(name='harry', pet_type='hamster')
使用哪种调用方式无关紧要,只要函数调用能生成希望的输出即可。
- 避免实参错误
提供的实参多于或少于函数完成其工作所需的信息时,将出现实参不匹配错误
返回值
函数并非总是直接显示输出,可以处理一些数据,并返回一个或一组值,称为返回值
。在函数中,可使用return
语句将值返回到调用函数的代码行。返回值可以将程序的大部分繁重工作移到函数中去完成,从而简化主程序。
- 简单返回值
def get_name(first_name, last_name):
"""返回整洁的姓名"""
full_name = first_name + ' ' + last_name
return full_name.title()
musician = get_name('jimi', 'hendrix')
print(musician)
函数get_name()
定义通过形参接受名和姓,并将整合后的结果返回到函数调用行。
-
让实参变成可选的
可选实参可以让使用函数的人只需要在必须时才提供额外的信息。可使用默认值来让参数变成可选的。 -
返回字典
函数可返回任何类型的值,包括列表和字典等较复杂的数据结构。
def get_person(first_name, last_name, age=''):
"""返回一个包含个人信息的字典"""
person = {'first': first_name, 'last': last_name}
if age:
person['age'] = age
return person
musician = get_person('jimi', 'hendrix', age='27')
print(musician)
在函数定义中新增了一个可选参数age
,并将其默认值设置为空字符串。
- 结合使用函数和while循环
def get_person(first_name, last_name):
"""返回整洁的姓名"""
full_name = first_name + ' ' + last_name
return full_name.title()
while True:
print('Please tell me your name!')
print("Please input 'q' to quit!")
f_name = input('First name: ')
if f_name == 'q':
break
l_name = input('Last name: ')
if l_name == 'q':
break
name = get_person(f_name, l_name)
print('Hello ' + name + '!')
传递列表
想函数传递列表,列表可包含的可能是名字、数字或更复杂的对象(如字典)。将列表传递给函数后,函数就能直接访问其内容。
假设有一个用户列表,要访问其中的每位用户:
def users(names):
"""向列表中的每位用户都发出简单的问候"""
for name in names:
msg = "Hello, " + name.title() + "!"
print(msg)
usernames = ['hannah', 'ty', 'margot']
users(usernames)
- 在函数中修改列表
将列表传递给函数后,函数就可以对其进行修改。在函数中对这个列表所做的任何修改都是永久性的,这能高效地处理大量的数据
def print_models(unprinted_designs, comleted_models):
"""
模拟打印每个设计,知道没有未打印设计为止
打印每个设计后,都将其添加到列表completed_models中
"""
while unprinted_designs:
current_design = unprinted_designs.pop()
# 模拟根据设计制作3D打印模型的过程
print("Printing model:" + current_design)
completed_models.append(current_design)
def show_completed_models(completed_models):
"""显示打印好的所有模型"""
print("\nThe following models have been printed:")
for completed_model in completed_models:
print(completed_model)
unprinted_designs = ['iphone case', 'robot pendant', 'dodecahedron']
completed_models = []
print_models(unprinted_designs, completed_models)
show_completed_models(completed_models)
将大部分工作代码都移到两个函数中。让主程序更容易理解,即每个函数都应只负责一项具体的工作,这将有助于将复杂的任务划分成一系列步骤。
- 禁止函数修改列表
使用切片表示法[ : ]
创建列表的副本。
print_models(unprinted_designs[:], completed_models)
这样函数print_models()
依旧能完成其工作,但是用的是列表unprinted_designs
的副本,而不是列表本身,所以函数所做的修改不会影响到列表。
虽然向函数传递列表的副本可保留原始列表的内容,但除非有充分的理由需要传递副本,否则还是应该将原始列表传递给函数。因为让函数使用现成列表可避免花时间和内存创建副本,从而提高效率,在处理大型列表时尤其如此。
传递任意数量的实参
在预先不知道函数需要接受多少实参时,Python允许函数从调用语句中收集任意数量的实参
例如,在无法确定顾客要什么配料的情况下制作比萨:
def make_pizza(*toppings):
"""打印顾客点的所有配料"""
for topping in toppings:
print('- ' + topping)
print('-'*20)
make_pizza('pepperoni')
make_pizza('mushrooms', 'green peppers', 'extra cheese')
形参名*toppings
中的星号*
让Python创建一个名为toppings的空元祖,并将收集到的所有值都封装到这个元祖中。
- 结合使用位置实参和任意数量实参
要让函数接受不同类型的实参,必须在函数定义中将接纳任意数量实参的形参放在最后。Python先匹配位置实参和关键字实参,再将余下的实参都收集到最后一个形参中。
def make_pizza(size, *toppings):
"""打印顾客点的所有配料"""
print("Make a " + str(size) + "-inch pizza.")
for topping in toppings:
print('- ' + topping)
print('-'*20)
make_pizza(13, 'pepperoni')
make_pizza(16, 'mushrooms', 'green peppers', 'extra cheese')
得到结果:
Make a 13-inch pizza.
- pepperoni
--------------------
Make a 16-inch pizza.
- mushrooms
- green peppers
- extra cheese
--------------------
- 使用任意数量的关键字实参
需要接受任意数量的实参,但预先不知道传递给函数的会是什么样的信息。在这种情况下,可将函数编写成能够任意接受数量的键-值
对——调用语句提供了多少就接受多少。
用户简介事例:知道将收到有关用户的信息,但不确定会是什么样的信息。函数build_profile()
接受名和姓,同时还接受任意数量的关键字实参:
def build_profile(first, last, **user_info):
"""创建一个字典,其中包含用户信息"""
profile = {'first_name': first, 'last_name': last}
for key, value in user_info.items():
profile[key] = value
return profile
user_profile = build_profile('albert', 'einstein',
location='princeton',
field='physics')
print(user_profile)
得到结果:
{'first_name': 'albert', 'last_name': 'einstein', 'location': 'princeton', 'field': 'physics'}
返回的字典包含用户的名和姓,还有求学的地方和所学专业。调用这个函数时,不管额外提供了多少键-值
对,他都能正确地处理。
编写函数时,可以以各种方式混合使用位置实参、关键字实参和任意数量的实参。知道这些实参类型大有裨益,因为阅读别人编写的代码时经常会见到它们。
匿名函数
Python支持匿名函数的使用。
- 匿名函数创建
匿名函数使用lambda
关键字创建。
lambda argument_list: expression
argument_list
argument_list
是参数列表,结构与Python中函数的参数列表是一样的expression
expression
是一个关于参数的表达式。表达式中出现的参数需要在argument_list中有定义,并且表达式只能是单行的。
在Python中,Lambda
表达式使用于简单的函数:
def square(x):
y = x**2
return y
上述square()
函数可以写成如下形式:
square = lambda x: x ** 2
注:PEP8中并不推荐用这种方式。
- 三个特性
lambda
函数是匿名的:没有名字的函数lambda
函数有输入和输出:输入是传入到参数列表argument_list
的值,输出是根据表达式expression
计算得到的值。lambda
函数一般功能简单:单行expression
决定了lambda
函数不可能完成复杂的逻辑,只能完成非常简单的功能。由于其实现的功能一目了然,甚至不需要专门的名字来说明。
- 用法
- 将
lambda
函数赋值给一个变量,通过这个变量间接调用该lambda
函数。
例如:square = lambda x: x ** 2
- 将
lambda
函数赋值给其他函数,从而将其他函数用该lambda
函数替换。
例如:time.sleep = lambda x: None
调用time.sleep()
时,不会执行任何操作 - 将
lambda
函数作为其他函数的返回值,返回给调用者
例如:
def add():
return lambda x, y: x+y
print(add()(1, 2))
lambda
函数实际上是定义在某个函数内部的函数,称之为嵌套函数,或者内部函数,内部函数能够访问外部函数的局部变量
- 将
lambda
函数作为参数传递给其他函数。
例如:map(lambda x: x+1, [1, 2, 3])
,将列表[1, 2, 3]
中的元素分别加1
将函数存储在模块中
通过进一步将函数存储在被称为模块的独立文件中,再将模块导入到主程序中。import
语句允许在当前运行的程序文件中使用模块中的代码。
将函数存储在被称为模块的独立文件中,再将模块导 入到主程序中。import语句允许在当前运行的程序文件中使用模块中的代码。
- 导入整个模块
要让函数是可导入的,得先创建模块。模块是扩展名为.py
的文件,包含要导入到程序中的代码。
将make_pizza()
函数之外的其他代码都删除,保存为pizza.py
文件,在同目录下创建另一个.py
文件:
import pizza
pizza.make_pizza(13, 'pepperoni')
pizza.make_pizza(16, 'mushrooms', 'green peppers', 'extra cheese')
代码行import pizza
让Python打开文件pizza.py
,并将其中的所有函数都复制到这个程序中。
- 导入特定的函数
可以导入模块中的特定函数,语法如下:
from module_name import function_0, function_1
通过逗号分隔函数名,可根据需要从模块中导入任意数量的函数
例如:
from pizza import make_pizza
pizza.make_pizza(13, 'pepperoni')
pizza.make_pizza(16, 'mushrooms', 'green peppers', 'extra cheese')
- 使用as给函数指定别名
如果要导入的函数的名称可能与程序中现有的名称冲突,或者函数的名称太长,可指定简短而独一无二的别名
——函数的另一个名称,类似于外号。
关键字as将函数重命名为你提供的别名:
from pizza import make_pizza as mp
- 使用as给模块指定别名
通过给模块指定简短的别名,能更轻松地调用模块中的函数。
import pizza as p
p.make_pizza(16, 'pepperoni')
p.make_pizza(12, 'mushrooms', 'green peppers', 'extra cheese')
- 导入模块中的所有函数
使用星号*
运算符可让Python导入模块中的所有函数
from pizza import *
make_pizza(16, 'pepperoni')
make_pizza(12, 'mushrooms', 'green peppers', 'extra cheese')
import
语句中的星号让Python将模块pizza
中的每个函数都复制到这个程序文件中。
使用并非自己编写的大型模块时,最好不要采用这种导入方法,如果模块中有函数的名称与当前项目中使用的名称相同,可能导致意想不到的结果,Python可能遇到多个名称相同的函数或变量,进而覆盖函数,而不是分别导入所有的函数。
最佳的做法是,要么只导入你需要使用的函数,要么导入整个模块并使用句点表示法。这能让代码更清晰,更容易阅读和理解。
函数编写指南
编写函数时,应给函数指定描述性名称,且只在其中使用小写字母
和下划线
。给模块命名时也应遵循上述约定。
每个函数都应包含简要地阐述其功能的注释,该注释应紧跟在函数定义后面,并采用文档字符串格式。只要知道函数的名称、需要的实参以及返回值的类型,就能轻松使用。
- 给形参指定默认值时,等号两边不要有空格
- 对于函数调用中的关键字实参,也应该遵守这种约定
- PEP8建议代码行的长度不要超过
79
字符。如果形参很多,导致函数定义的长度超过79
字符,可在函数定义中输入左括号后按回车键
,并在下一行按两次Tab
键,从而将形参列表和只缩进一层的函数体区分开来。 - 如果程序或模块包含多个函数,可使用两个空行将相邻的函数分开
- 所有
import
语句都应该放在文件开头,唯一例外是在文件开头使用了注释来描述整个程序