Python函数和模块的使用

我的博客

文章首发于公众号：小肖学数据分析

在开发过程中，函数和模块帮助我们将复杂的代码逻辑分解为可管理的部分，提升代码的可读性、可维护性和重用性。

本文将介绍如何在Python中有效利用函数和模块，提供详细的示例。

函数的作用

在多次执行相同或相似代码逻辑时，如果不使用函数，代码会显得冗长且重复。通过函数封装功能模块，我们可以避免这种难以维护和理解的代码。

定义函数

在Python中，我们用def关键字来定义一个函数。

# 使用def定义一个函数，计算平方
def square(number):
    # 返回值为参数的平方
    return number * number

# 调用自定义函数
result = square(5)  # result的值为25

调用函数

Python有很多内置函数，可以直接调用它们，也可以通过导入模块来扩展这些功能。

# 导入math模块以使用数学相关的函数
import math

# 调用math模块中的sqrt函数计算平方根
root = math.sqrt(25)  # root的值为5.0

函数的参数

函数参数的灵活性是Python函数的一大特点，包括默认参数、可变参数、关键字参数和命名关键字参数。

默认参数 (Default Arguments)

默认参数为函数的某些参数指定默认值。

# 定义一个函数，其中一个参数有默认值
def make_coffee(size, sugar=2, milk=True):
    # 根据参数制作咖啡
    coffee = f"{size} coffee"
    coffee += f" with {sugar} sugars" if sugar else " with no sugar"
    coffee += " and milk" if milk else " without milk"
    return coffee

print(make_coffee("large"))  # 使用默认的糖量和牛奶
print(make_coffee("medium", milk=False))  # 不用牛奶
print(make_coffee("small", sugar=0))  # 不加糖

可变参数 (*args)

可变参数允许函数接收任意数量的参数，这些参数被封装进一个元组。

# 定义一个函数，接收任意数量的参数
def make_sandwich(bread, *fillings):
    # 组合三明治的配料
    sandwich = f"{bread} sandwich with"
    for filling in fillings:
        sandwich += f" {filling},"
    return sandwich[:-1]  # 移除最后的逗号

print(make_sandwich("rye", "cheese", "ham", "tomato"))  # 多个填充
print(make_sandwich("whole grain"))  # 无填充

关键字参数 (**kwargs)

通过关键字参数，可以接收一个包含所有未显式接收的关键字参数的字典。

# 定义一个函数，接收任意的键值对作为参数
def make_pizza(size, **toppings):
    # 根据键值对准备披萨的配料
    pizza = f"{size} pizza with"
    for topping, value in toppings.items():
        pizza += f" {value} {topping},"
    return pizza[:-1]  # 移除最后的逗号

print(make_pizza("large", cheese="extra", pepperoni="double", olives="yes"))

命名关键字参数

命名关键字参数要求调用者必须使用参数的名字来指定它们。

# 定义一个函数，使用命名关键字参数
def create_user(*, username, password, email):
    # 根据提供的名字关键字参数创建一个用户
    return {"username": username, "password": password, "email": email}

# 正确使用命名关键字参数
user = create_user(username="alice", password="secret", email="alice@example.com")
print(user)

参数组合

在Python函数中，可以组合使用上述各种类型的参数。

# 定义一个函数，结合使用不同类型的参数
def create_profile(name, age, *hobbies, **details):
    # 创建用户档案
    profile = {"name": name, "age": age, "hobbies": hobbies}
    profile.update(details)
    return profile

# 使用组合参数创建档案
profile = create_profile("Alice", 28, "hiking", "cooking", city="Seattle", email="alice@example.com")
print(profile)

函数的返回值

函数不仅可以做计算还可以返回值，包括没有返回值、返回单个值或返回多个值。

# 一个返回多个值的函数
def get_stats(numbers):
    # 返回最大值和最小值
    return max(numbers), min(numbers)

# 接收返回的多个值
max_val, min_val = get_stats([1, 2, 3, 4, 5])  # max_val为5，min_val为1

作用域问题

理解Python中的作用域概念对于编写正确的代码至关重要。

# 全局变量
x = 10

def change_value():
    # 局部变量
    x = 5
    print("Inner x:", x)

change_value()  # 输出: Inner x: 5
print("Outer x:", x)  # 输出: Outer x: 10

用模块管理函数

把相关的函数组织在一起形成模块，可以帮助我们更好地组织代码结构。

# 假设这是一个名为mymodule.py的文件
def add(a, b):
    return a + b

def subtract(a, b):
    return a - b

# 在其他Python文件中使用这个模块
import mymodule

# 使用模块中定义的函数
result = mymodule.add(3, 4)  # result的值为7

当遇到命名冲突时，Python会根据作用域规则来决定使用哪个变量。同一个模块内的命名冲突通常会在加载时被覆盖，而不同模块间的冲突可以通过模块名来区分。