functools 是 Python 标准库中一个极为实用的模块,提供了许多用于处理函数、高阶函数以及函数式编程的工具。
1. functools.partial
用途: 创建一个新函数,它固定了原函数的部分参数。
示例:
from functools import partial
def power(base, exponent):
return base ** exponent
# 创建一个固定底数为2的新函数
square = partial(power, base=2)
# 调用新函数只需传递指数
print(square(3)) # 输出: 8
2. functools.update_wrapper / functools.wraps
用途: 在编写装饰器时,保持被装饰函数的元信息(如名称、文档字符串、注解等)不变。
示例:
from functools import wraps
def add_logging(func):
@wraps(func)
def wrapper(*args, **kwargs):
print(f"Calling {func.__name__} with arguments {args}, {kwargs}")
result = func(*args, **kwargs)
print(f"{func.__name__} returned: {result}")
return result
return wrapper
@add_logging
def greet(name):
"""Greet someone by their name."""
return f"Hello, {name}!"
print(greet.__name__) # 输出: greet
print(greet.__doc__) # 输出: Greet someone by their name.
greet("Alice") # 输出: Calling greet with arguments ('Alice',), {} 和 "greet returned: Hello, Alice!"
3. functools.total_ordering
用途: 通过实现一个或少数几个关键比较方法(如 __eq__ 和 __lt__),自动为一个类生成所有必要的比较方法(如 __le__, __gt__, __ge__, __ne__)。
示例:
from functools import total_ordering
@total_ordering
class Person:
def __init__(self, name, age):
self.name = name
self.age = age
def __eq__(self, other):
return self.name == other.name and self.age == other.age
def __lt__(self, other):
return self.age < other.age
# 自动拥有所有比较方法,如 __le__, __gt__, __ge__, __ne__
p1 = Person("Alice", 25)
p2 = Person("Bob", 30)
print(p1 <= p2) # 输出: True
print(p1 > p2) # 输出: False
4. functools.lru_cache
用途: 实现函数结果的缓存,以提高性能。基于“最近最少使用”(Least Recently Used, LRU)策略自动丢弃旧的缓存项。
示例:
from functools import lru_cache
@lru_cache(maxsize=32)
def fibonacci(n):
if n < 2:
return n
return fibonacci(n - 1) + fibonacci(n - 2)
# 多次调用相同参数时,从缓存中获取结果,避免重复计算
print(fibonacci(20)) # 第一次调用进行计算
print(fibonacci(20)) # 第二次调用直接使用缓存
5. functools.reduce
用途: 对一个可迭代对象(如列表)应用一个累积函数,从左到右将元素两两结合,返回单一结果。
示例:
from functools import reduce
numbers = [1, 2, 3, 4, 5]
# 计算列表元素的乘积
product = reduce(lambda x, y: x * y, numbers)
print(product) # 输出: 120
6. functools.cmp_to_key
用途: 将旧式的比较函数(接受两个参数并返回负数、零或正数表示小于、等于或大于)转换为可用于排序的键函数。
示例:
from functools import cmp_to_key
def compare_names(name1, name2):
if name1 < name2:
return -1
elif name1 > name2:
return 1
else:
return 0
names = ["Alice", "Bob", "Charlie", "David"]
# 使用旧式比较函数进行排序
sorted_names = sorted(names, key=cmp_to_key(compare_names))
print(sorted_names) # 输出: ['Alice', 'Bob', 'Charlie', 'David']
以上示例
涵盖了 functools 模块中一些常用且重要的功能,包括部分函数应用、装饰器元信息保留、简化类比较方法、函数缓存、累积计算以及旧式比较函数的转换。根据实际需求,可以灵活运用这些工具来增强代码的简洁性、效率和可读性。