你经常会发现,向函数传递列表很有用。这种列表包含的可能是名字,数字或更复杂的对象(如字典)。将列表传递给函数后,函数就能直接访问其内容。
假设有一个用户列表,我们要问候其中的每一位用户。下面的示例将一个名字列表传递给一个名为greet_users()的函数,这个函数问候列表中的每个人:
def greet_users(names): for name in names: msg = "Hello, " + name.title() + "!" print(msg) usernames = ['hannah','ty','margot'] greet_users(usernames)
我们将greet_users()定义成接受一个名字列表,并将其存储在形参names中。这个函数遍历收到的列表,并对其中每位用户都打一句问候语。
Hello, Hannah!
Hello, Ty!
Hello, Margot!
在函数中修改列表
将列表传递给函数后,函数就可对其进行修改。在函数中对这个列表所做的任何修改都是永久性的,这让你能高效地处理大量地数据。
来看一家为用户提交的设计制作3D打印模型的公司。需要打印的设计存储在一个列表中,打印后移动到另一个列表中。
#首先创建一个列表,其中包含一些要打印的设计 unprinted_designs = ['iphone case','robot pendant','dodecahedron'] completed_models = [] #模拟打印每个设计,直到没有未打印的设计为止 #打印每个设计后,都将其移到列表completed_models中 while unprinted_designs: current_design = unprinted_designs.pop() #模拟根据设计制作3D打印模型的过程 print("Printing model: " + current_design) completed_models.append(current_design) #显示打印好的所有模型 print(" The following models have been printed:") for completed_model in completed_models: print(completed_model)
这个程序首先创建一个需要打印的设计列表,还创建一个空列表,每个设计打印都将移到这个列表中。
Printing model: dodecahedron
Printing model: robot pendant
Printing model: iphone case
The following models have been printed:
dodecahedron
robot pendant
iphone case
为重新组织这些代码,我们可编写两个函数,每一个都做一件具体的工作。大部分代码都与原来相同,只是效率更高。第一个函数将负责处理打印设计的工作,而第二个将概述打印了哪些设计:
def print_models(unprinted_designs,completed_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(" The 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(),它包含两个形参:一个需要打印的设计列表和一个打印好的模型列表。给定这两个列表,这个函数模拟打印每个设计的过程:将设计逐个地从未打印的设计列表中取出,并加入打印好的模型列表中。
我们又定义了函数show_completed_models(),它包含一个形参:打印好的模型列表。给定这个列表,函数将显示打印出来的每个模型的名称。
Printing model: dodecahedron
Printing model: robot pendant
Printing model: iphone case
The following models have been printed:
dodecahedron
robot pendant
iphone case
相对没有使用函数的版本,这个程序更容易扩展和维护。如果以后需要打印其他设计,只需要再次调用print_models()即可。
禁止函数修改列表
有时候,需要禁止函数修改列表。例如,假设像前一个示例那样,你有一个未打印的设计列表,并编写了一个将这些设计移到打印好的模型列表中的函数。你可能会做出这样的决定:即使打印了所有设计后,也要保留原来的未打印的设计列表。但由于你将所有的设计都移出了unprinted_designs,这个列表变成了空的,原来的列表没有了。但为了解决这个问题,可向函数传递列表的副本而不是原件:这样函数所做的任何修改都只影响副本,而不影响原件。
要将列表的副本传递给函数,可以像下面这样做:
function_name(list_name[:])
切片表示法[:]创建列表的副本。在print_models.py中,如果不想清空未打印的设计列表,可像下面这样调用print_models():
print_models(unprinted_designs[:],completed_models)
这样函数print_models()依然能够完成其工作,但它使用的是列表unprinted_designs的副本,而不是其本身。