如何在Python的sorted方法中为key函数传递额外参数？

你遇到的问题核心是：sorted的key参数要求是一个仅接收单个元素参数的可调用对象，而你直接调用compare_function(cls_name)会立即执行函数并返回结果（不是函数本身），导致obj参数缺失；直接写compare_function(obj, cls_name)则因为obj未在当前作用域定义而报错。

下面是三种通用解决方案：

1. 使用lambda表达式包装（最常用） #

用lambda接收sorted传入的obj，再把obj和额外参数一起传给你的compare_function：

if order == "dsc":
    return sorted(qs, key=lambda obj: compare_function(obj, cls_name), reverse=True)
elif order == "asc":
    return sorted(qs, key=lambda obj: compare_function(obj, cls_name))
else:
    pass

def compare_function(obj, cls_name):
    if obj is None:     
        return "aaaaaaa"
    if obj.teacher is None: 
        return "aaaaaaa"
    else:   
        # 此处可使用传入的cls_name执行自定义逻辑
        return obj.teacher.name.lower()

2. 使用functools.partial绑定参数 #

functools.partial可以预先绑定函数的部分参数，生成一个仅需接收obj的新函数：

from functools import partial

if order == "dsc":
    key_func = partial(compare_function, cls_name=cls_name)
    return sorted(qs, key=key_func, reverse=True)
elif order == "asc":
    key_func = partial(compare_function, cls_name=cls_name)
    return sorted(qs, key=key_func)
else:
    pass

def compare_function(obj, cls_name):
    if obj is None:     
        return "aaaaaaa"
    if obj.teacher is None: 
        return "aaaaaaa"
    else:   
        # 使用cls_name执行自定义逻辑
        return obj.teacher.name.lower()

3. 使用闭包捕获外部参数 #

把compare_function定义在一个外层函数中，让它直接捕获外层作用域的cls_name，无需额外传参：

def get_key_func(cls_name):
    def compare_function(obj):
        if obj is None:     
            return "aaaaaaa"
        if obj.teacher is None: 
            return "aaaaaaa"
        else:   
            # 直接使用外层的cls_name
            return obj.teacher.name.lower()
    return compare_function

# 调用逻辑
if order == "dsc":
    key_func = get_key_func(cls_name)
    return sorted(qs, key=key_func, reverse=True)
elif order == "asc":
    key_func = get_key_func(cls_name)
    return sorted(qs, key=key_func)
else:
    pass

注意点 #

• 建议用is None代替== None，这是Python的最佳实践。
• 三种方案各有适用场景：lambda最简洁，适合简单场景；partial适合参数较多或需复用绑定后函数的场景；闭包适合封装复杂逻辑的场景。

内容的提问来源于stack exchange，提问作者Alston