Question

我知道可以使用__repr__魔术方法来自定义对象的表示形式。

class A:

    ...

    def __repr__(self):
        return 'A(id={})'.format(self.id)

当它们是这些对象的列表时，是否可以自定义表示，因此它不只是打印对象的表示列表[A(id=1), A(id=10), A(id=11)]。

例如：

class AGroup:
    def __init__(self, lst):
        self.lst = lst

>>> lst = [A(1), A(10), A(11)]
>>> grp = AGroup(lst)
>>> grp.lst
<Group of 3 A's>  # instead of [A(id=1), A(id=10), A(id=11)]

我正在考虑将列表包装到另一个对象中，然后再分配给self.lst，以便我可以修改其代表。还有其他想法吗？

Answer 1

您可以将传递给AGroup构造函数的列表存储在list的子类的实例中，以便您可以覆盖{{1}的__repr__方法}子类以您想要的格式返回输出：

list

这将输出：

class A:
    def __init__(self, id):
        self.id = id
    def __repr__(self):
        return 'A(id={})'.format(self.id)

class AGroup:
    class AList(list):
        def __repr__(self):
            return "<Group of %d A's>" % len(self)

    def __init__(self, lst):
        self.lst = self.AList(lst)

lst = [A(1), A(10), A(11)]
grp = AGroup(lst)
print(grp.lst)

Answer 2

要分组类实例，我们需要覆盖eq，ne和hash函数。通过提供静态哈希值，会将此类的所有实例视为一个对象。

from collections import Counter
class A:
    def __init__(self , id):
        self.id = id
    def __hash__(self): return hash('A')
    def __eq__(self, x): return type(x) is A
    def __ne__(self, x): return type(x) is not A

    def __repr__(self):
        return 'A'


class B:
    def __init__(self , id):
        self.id = id
    def __hash__(self): return hash('B')
    def __eq__(self, x): return type(x) is B
    def __ne__(self, x): return type(x) is not B

    def __repr__(self):
        return 'B'


class AGroup:
    def __init__(self, lst):
        count = Counter(lst)
        self.lst = []
        for cl in Counter(lst):
            self.lst.append('<Group of {} {}\'s>'.format(count[cl] , cl))



lst = [A(1), B(10), A(11)]
grp = AGroup(lst)
grp.lst


["<Group of 2 A's>", "<Group of 1 B's>"]

对象列表的自定义列表表示

2 个答案: