Python operator.attrgetter 怎样批量读取对象属性

在处理一组对象时，如果排序、字段投影和分组都反复写 lambda item: item.name，属性读取逻辑很快会散落在多处。Python 的 operator.attrgetter 可以把属性路径预先做成一个可调用对象，再交给 sorted()、map() 或其他接收函数参数的工具。

直接结论：单个属性用 attrgetter("name") 返回标量；多个属性用 attrgetter("name", "score") 返回按声明顺序排列的元组；点号路径如 attrgetter("customer.profile.display_name") 会逐层取属性，中间任一层不存在就会抛出 AttributeError。

官方地址：https://docs.python.org/3/library/operator.html#operator.attrgetter

先把重复的 lambda 收拢成一个访问器

attrgetter 返回的不是读取结果，而是一个可以反复调用的函数对象。把它保存到变量后，既可以直接调用，也可以作为排序键或映射函数传入。

from dataclasses import dataclass
from operator import attrgetter

@dataclass
class Student:
    name: str
    score: int

students = [
    Student("林一", 86),
    Student("周宁", 93),
    Student("陈默", 78),
]

# 先创建可复用的属性访问器，避免每个调用点重复写 lambda。
get_name = attrgetter("name")

# sorted 会把每个 Student 交给 get_name，并按返回的字符串排序。
by_name = sorted(students, key=get_name)

# map 复用同一个访问器，得到名字序列；这里显式转 list 便于后续使用。
names = list(map(get_name, students))

# 访问器本身仍可直接调用，参数必须是拥有 name 属性的对象。
first_name = get_name(students[0])

这里的关键是“把字段读取当作函数传递”。sorted 负责遍历对象，get_name 只负责返回排序键，二者职责清晰。若只是一次性的简单排序，lambda 仍然足够；当同一字段要被排序、投影和分组多次使用时，attrgetter 更容易集中管理。

单属性和多属性的返回形状不同

单属性访问器返回该属性的原值，多属性访问器返回元组。这个差异会影响解包、排序键和后续的类型判断，不能只把两种写法当成参数数量不同。

from operator import attrgetter

# 单属性：返回一个字符串，不会自动包装成一元组。
get_name = attrgetter("name")
name = get_name(Student("林一", 86))

# 多属性：返回元组，元组顺序与 attrgetter 的参数顺序一致。
get_name_score = attrgetter("name", "score")
name_score = get_name_score(Student("林一", 86))
name_value, score_value = name_score

# 多字段元组也可以直接作为排序键，先按 name，再按 score。
ordered = sorted(students, key=attrgetter("name", "score"))

多属性返回元组后，Python 的元组比较会按位置进行比较。因此 sorted(..., key=attrgetter("name", "score")) 的含义是先比较姓名，姓名相同时再比较分数。若业务要求先按分数降序、再按姓名升序，建议把两个排序条件写成清楚的排序策略，不要误以为 attrgetter 能为每个字段单独配置方向。

点号路径适合读取嵌套对象

属性名可以包含点号，attrgetter 会按照点号切分后逐层访问。它适合对象结构稳定、字段链路明确的场景，例如从订单对象取客户展示名。

from dataclasses import dataclass
from operator import attrgetter

@dataclass
class Profile:
    display_name: str

@dataclass
class Customer:
    profile: Profile

@dataclass
class Order:
    order_id: str
    customer: Customer

orders = [
    Order("A-102", Customer(Profile("周宁"))),
    Order("A-101", Customer(Profile("林一"))),
]

# 点号路径表示连续的属性读取，不是字典键路径。
get_display_name = attrgetter("customer.profile.display_name")

# 同一个访问器可用于排序和字段投影。
orders_by_customer = sorted(orders, key=get_display_name)
customer_names = list(map(get_display_name, orders))

这里的 customer.profile.display_name 是对象属性链。如果 customer 为 None，或者某一层没有目标属性，访问就会在对应位置失败。attrgetter 不会把缺失属性自动转换成 None，也不会替业务决定默认值。

缺失属性时要决定是暴露异常还是先做清洗

如果数据契约要求每个对象都有完整属性，直接让 AttributeError 暴露出来通常更好，它能尽早发现脏数据或对象类型混入。若输入允许缺少字段，则应在进入 attrgetter 前完成归一化，或者使用带默认值的显式函数。

from operator import attrgetter

get_name = attrgetter("name")

def safe_name(item):
    # 只有业务明确允许缺失 name 时才提供默认值，避免吞掉结构错误。
    try:
        return get_name(item)
    except AttributeError:
        return "未命名"

# 先统一结果，再交给排序；这样排序逻辑不需要处理异常。
ordered = sorted(students, key=safe_name)

不要为了“让程序继续跑”而无条件捕获所有异常。AttributeError 可能来自嵌套链路中的任何一层，生产代码最好在输入边界记录对象类型和字段来源，再决定是拒绝数据、补齐字段还是使用默认值。

attrgetter、getattr 和 itemgetter 怎么选

一个简单判断方法是：数据字段写成 obj.name 就优先考虑 attrgetter，写成 obj["name"] 就考虑 itemgetter；属性名来自配置且需要默认值时，再使用显式的 getattr(obj, field, default)。

把这套写法放进实际工作流

1. 先确认输入是对象属性还是字典/序列下标。
2. 字段固定且会被多处消费时，创建一个命名清楚的 attrgetter。
3. 单字段直接使用标量结果，多字段按声明顺序解包元组。
4. 嵌套路径只用于结构稳定的对象，明确中途缺失时的异常策略。
5. 将访问器交给 sorted、map 或 itertools.groupby 前，先约定返回值形状。

掌握这几个边界后，operator.attrgetter 的价值不在于少写几个字符，而在于把“如何取字段”从排序、投影等业务动作中分离出来，让同一个字段访问规则可以被重复使用。

常见问题

attrgetter 可以读取字典键吗？

不能直接把字典键当成对象属性。字典场景应使用 operator.itemgetter("name")，或者用 getattr 处理真正的对象属性。

多个属性的返回值一定是元组吗？

传入两个或更多属性名时返回元组，顺序与参数顺序一致；只传一个属性名时返回该属性本身，而不是一元组。

如何给不存在的嵌套属性设置默认值？

attrgetter 没有默认值参数。可以在数据清洗阶段补齐对象，或写一个明确捕获 AttributeError 的包装函数，只对业务允许缺失的字段提供默认值。