Python中自定义对象排序键的灵活配置与封装实践

admin 百科 2025-12-12 12

本教程探讨了在python中对自定义类对象进行排序时，如何通过lambda函数灵活指定排序键，并解决了常见的typeerror: 'person' object is not subscriptable问题。文章将展示如何正确访问对象属性作为排序依据，并讨论了在封装性要求下的最佳实践，包括直接属性访问、引入getter方法以及使用operator.attrgetter。

在Python编程中，对包含自定义对象（如类实例）的列表进行排序是一个常见需求。无论是使用Python内置的list.sort()方法、sorted()函数，还是自定义的排序算法（例如归并排序），都可以通过key参数来指定排序依据。key参数接受一个可调用对象（通常是lambda函数），该对象将作用于列表中的每个元素，并返回一个用于比较的值。

问题背景：自定义对象排序中的类型错误

考虑一个Person类，它包含姓名、年龄、身高和体重等属性。我们希望能够根据这些不同的属性对Person对象列表进行排序。

以下是初始的Person类定义和尝试排序的代码片段：

import numpy as np
import enum
# 假设 merge_sort 模块中包含 mergesort 函数
# from merge_sort import mergesort 

NAMES = ["Alice", "Bob", "Charlie", "David", "Eve", "Frank", "Grace", "Hank", "Ivy", "Jack"]

class Person():
    def __init__(self, name, age, height, weight):
        self._name = name
        self._age = age
        self._height = height
        self._weight = weight

    def __repr__(self):
        return f"Person('{self._name}', {self._age}, {self._height}, {self._weight})"

    # 其他魔术方法 __eq__, __lt__ 等略

def create_persons_list(n=10, sort_key='weight'):
    person_objects = [Person(np.random.choice(NAMES), np.random.randint(18, 101), 
                             np.random.randint(150, 201), np.random.randint(45, 101)) 
                      for _ in range(n)]

    # 尝试使用索引作为排序键，导致TypeError
    if sort_key == 'name':
        # return mergesort(person_objects, key=lambda x: x[0]) 
        pass # 此处会报错
    # ... 其他条件分支

# 假设 mergesort 存在并能正确工作，这里使用 sorted 作为替代演示
def mergesort(data, key):
    # 实际的归并排序实现，这里仅为演示 key 参数作用
    return sorted(data, key=key)

# 示例调用 (会导致TypeError)
# sorted_persons_by_name = create_persons_list(sort_key='name')

登录后复制

当尝试运行类似 mergesort(person_objects, key=lambda x: x[0]) 的代码时，Python会抛出 TypeError: 'Person' object is not subscriptable。这个错误的原因在于，lambda函数中的参数 x 是一个 Person 类的实例，而不是一个像列表或元组那样可以通过索引（[0]）来访问元素的序列。Person类默认情况下不支持下标操作，因此尝试对其进行下标访问会导致类型错误。

立即学习“Python免费学习笔记（深入）”；

解决方案：直接访问对象属性

要解决这个问题，我们需要理解 lambda 函数接收的是 Person 对象本身。因此，正确的做法是直接通过属性名访问 Person 对象的相应属性作为排序键。例如，如果我们要按年龄排序，就应该访问 x._age。

修改后的 create_persons_list 函数如下所示：

import numpy as np
# from merge_sort import mergesort # 假设 mergesort 函数可用

NAMES = ["Alice", "Bob", "Charlie", "David", "Eve", "Frank", "Grace", "Hank", "Ivy", "Jack"]

class Person():
    def __init__(self, name, age, height, weight):
        self._name = name
        self._age = age
        self._height = height
        self._weight = weight

    def __repr__(self):
        # 优化 repr，使其更具可读性
        return f"Person(name='{self._name}', age={self._age}, height={self._height}, weight={self._weight})"

    # 为了使 Person 对象能够被直接比较，可以实现比较魔术方法
    # 注意：这些方法定义了 Person 对象之间的默认比较行为，
    # 而 key 参数允许我们基于特定属性进行排序，两者作用不同。
    def __eq__(self, other):
        if not isinstance(other, Person):
            return NotImplemented
        return (self._age, self._height, self._weight) == (other._age, other._height, other._weight)

    def __lt__(self, other):
        if not isinstance(other, Person):
            return NotImplemented
        return (self._age, self._height, self._weight) < (other._age, other._height, other._weight)

    # ... 其他 __le__, __gt__, __ge__ 方法类似

# 假设 mergesort 存在并能正确工作，这里使用 sorted 作为替代演示
def mergesort(data, key):
    # 实际的归并排序实现，这里仅为演示 key 参数作用
    return sorted(data, key=key)

def create_persons_list(n=10, sort_key='weight'):
    person_objects = [Person(np.random.choice(NAMES), np.random.randint(18, 101), 
                             np.random.randint(150, 201), np.random.randint(45, 101)) 
                      for _ in range(n)]

    # 使用字典映射排序键到对应的 lambda 函数，提高代码可读性和可维护性
    key_map = {
        'name': lambda x: x._name,
        'age': lambda x: x._age,
        'height': lambda x: x._height,
        'weight': lambda x: x._weight,
    }

    if sort_key in key_map:
        return mergesort(person_objects, key=key_map[sort_key])
    else:
        raise ValueError("Invalid sort_key. Supported values are 'name', 'age', 'height', and 'weight'.")

# 示例调用
print("Sorted by name: ")
sorted_persons_by_name = create_persons_list(sort_key='name')
for p in sorted_persons_by_name:
    print(p)

print("\nSorted by age: ")
sorted_persons_by_age = create_persons_list(sort_key='age')
for p in sorted_persons_by_age:
    print(p)

登录后复制

Python中自定义对象排序键的灵活配置与封装实践-第2张图片-佛山资讯网