Python 函数核心指南：参数传递、返回值与模块使用

def describe_pet(pet_name, animal_type="dog"):
    print(f"\nI have a {animal_type}.")
    print(f"My {animal_type}'s name is {pet_name.title()}.")

# 只传一个参数，animal_type 使用默认值
describe_pet("xiaobai")
# 覆盖默认值
describe_pet("xiaohei", "cat")

返回值

使用 return 语句将结果返回给调用处。函数可以返回任意类型的数据，包括简单值、列表或字典。

返回简单值

def get_formatted_name(first_name, last_name):
    """返回整洁的姓名"""
    full_name = first_name + ' ' + last_name
    return full_name.title()

result = get_formatted_name('zhang', 'san')
print(result)

让实参变成可选的

通过默认值处理可选参数，避免报错。

def get_formatted_name(first_name, last_name, middle_name=''):
    """返回整洁的姓名"""
    if middle_name:
        full_name = first_name + ' ' + middle_name + ' ' + last_name
    else:
        full_name = first_name + ' ' + last_name
    return full_name.title()

print(get_formatted_name("zhang", "san"))
print(get_formatted_name("zhang", "san", "da"))

返回字典

常用于构建对象结构。

def build_person(first_name, last_name, age=''):
    """返回包含人员信息的字典"""
    person = {'first': first_name, 'last': last_name}
    if age:
        person['age'] = age
    return person

result = build_person('li', 'si')
print(result)

结合循环使用

将函数与 while 循环结合，可实现交互式输入。

def get_formatted_name(first_name, last_name):
    full_name = first_name + ' ' + last_name
    return full_name.title()

while True:
    print("\nPlease tell me your name:")
    f_name = input("First name: ")
    if f_name == 'q': break
    l_name = input("Last name: ")
    if l_name == 'q': break
    formatted_name = get_formatted_name(f_name, l_name)
    print(f"\nHello, {formatted_name}!")

传递列表

将列表传递给函数后，函数可直接访问并修改其内容。这种修改是永久性的。

遍历列表

def greet_users(names):
    """向列表中的每位用户发出简单的问候"""
    for name in names:
        msg = f"Hello, {name.title()}!"
        print(msg)

usernames = ['zhang san', 'li si', 'wang wu']
greet_users(usernames)

修改列表与副本

如果在函数中修改了传入的列表，原列表也会改变。若需保留原列表，应传递副本。

def print_models(unprinted_designs, completed_models):
    while unprinted_designs:
        current_design = unprinted_designs.pop()
        print(f"Printing model: {current_design}")
        completed_models.append(current_design)

def show_completed_models(completed_models):
    print("\nThe following models have been printed:")
    for model in completed_models:
        print(model)

unprinted_designs = ['iphone case', 'robot pendant']
completed_models = []

# 传递副本 [:] 以避免修改原列表
print_models(unprinted_designs[:], completed_models)
show_completed_models(completed_models)

传递任意数量的实参

可变位置实参 (*args)

形参前加星号 *，接收到的实参将被封装成元组。

def make_pizza(size, *toppings):
    """概述要制作的比萨"""
    print(f"\nMaking a {size}-inch pizza with the following toppings:")
    for topping in toppings:
        print(f"- {topping}")

make_pizza(16, "pepperoni", "mushrooms")

可变关键字实参 (**kwargs)

形参前加双星号 **，接收到的实参将被封装成字典。

def build_profile(first, last, **user_info):
    """创建一个字典，其中包含我们知道的有关用户的一切"""
    profile = {}
    profile['first_name'] = first
    profile['last_name'] = last
    for key, value in user_info.items():
        profile[key] = value
    return profile

user_profile = build_profile('zhang', 'san', location='xisanqi', field='big data')
print(user_profile)

将函数存储在模块中

将函数存储在独立的 .py 文件中即为模块。导入模块可使用以下语法：

导入整个模块

import module_name
module_name.function_name(args)

导入特定函数

from module_name import function_name
function_name(args)

指定别名

为避免命名冲突或简化调用，可使用 as 指定别名。

from module_name import function_name as fn
fn(args)

import module_name as mn
mn.function_name(args)

注意：除非自己编写的模块，否则不要轻易使用 from module_name import *，这可能导致命名冲突。