Python 3 Sortierung ignoriert __lt__ und __eq__

Question

Ich verwende Python 3.5.1 auf 64-Bit-Windows. Mein Problem ist Python scheint __eq__ und __lt__ Operatoren auf einer benutzerdefinierten Klasse beim Sortieren zu ignorieren. Die Verwendung eines benutzerdefinierten Sortierschlüssels ist eine Problemumgehung, die nicht funktioniert beim Versuch, Tupel zu sortieren, die Instanzen dieser Klasse enthalten.

Beispiel:

class Symbol: 
    def __init__(self, name, is_terminal = False): 
    self.name = name 
    self.is_terminal = is_terminal 
    def __eq__(self, other): 
    return (self.is_terminal, self.name) == (other.is_terminal, other.name) 
    def __lt__(self, other): 
    return (self.is_terminal, self.name) < (other.is_terminal, other.name) 

symbols = set() 
for s in "abcdef": 
    symbols.add(Symbol(s)) 

sorted_symbols = sorted(symbols) 
# sorted_symbols now contain the symbols in random order 

Mit dem functools.total_ordering Dekorateur nicht hilft.

Meine Frage ist, wie definiere ich die Bestellung für eine Benutzerklasse in Python 3?

Answer 1

Python tut nicht ignorieren __eq__ und __lt__, zumindest nicht, wenn Sie @functools.total_ordering tatsächlich verwendet:

>>> from functools import total_ordering 
>>> @total_ordering 
... class Symbol: 
...  def __init__(self, name, is_terminal=False): 
...   self.name = name 
...   self.is_terminal = is_terminal 
...  def __repr__(self): 
...   return 'Symbol({0.name!r}, is_terminal={0.is_terminal!r})'.format(self) 
...  def __hash__(self): 
...   return hash(self.name)^hash(self.is_terminal) 
...  def __eq__(self, other): 
...   print('{} __eq__ {}'.format(self, other)) 
...   return (self.is_terminal, self.name) == (other.is_terminal, other.name) 
...  def __lt__(self, other): 
...   print('{} __lt__ {}'.format(self, other)) 
...   return (self.is_terminal, self.name) < (other.is_terminal, other.name) 
... 
>>> symbols = set() 
>>> for s in "abcdef": 
...  symbols.add(Symbol(s)) 
... 
>>> sorted(symbols) 
Symbol('f', is_terminal=False) __lt__ Symbol('c', is_terminal=False) 
Symbol('a', is_terminal=False) __lt__ Symbol('f', is_terminal=False) 
Symbol('a', is_terminal=False) __lt__ Symbol('f', is_terminal=False) 
Symbol('a', is_terminal=False) __lt__ Symbol('c', is_terminal=False) 
Symbol('b', is_terminal=False) __lt__ Symbol('c', is_terminal=False) 
Symbol('b', is_terminal=False) __lt__ Symbol('a', is_terminal=False) 
Symbol('d', is_terminal=False) __lt__ Symbol('c', is_terminal=False) 
Symbol('d', is_terminal=False) __lt__ Symbol('f', is_terminal=False) 
Symbol('e', is_terminal=False) __lt__ Symbol('c', is_terminal=False) 
Symbol('e', is_terminal=False) __lt__ Symbol('f', is_terminal=False) 
Symbol('e', is_terminal=False) __lt__ Symbol('d', is_terminal=False) 
[Symbol('a', is_terminal=False), Symbol('b', is_terminal=False), Symbol('c', is_terminal=False), Symbol('d', is_terminal=False), Symbol('e', is_terminal=False), Symbol('f', is_terminal=False)] 

In der Tat, Sortierung funktioniert auch ohne @total_ordering weil die Timsort Implementierung nur __lt__ für diesen Satz verwendet.

Ein Sortierschlüssel ist auch eine Option, geben Sie einfach die (is_terminal, name) Tupel aus dem Schlüssel:

>>> sorted(symbols, key=lambda s: (s.is_terminal, s.name)) 
[Symbol('a', is_terminal=False), Symbol('b', is_terminal=False), Symbol('c', is_terminal=False), Symbol('d', is_terminal=False), Symbol('e', is_terminal=False), Symbol('f', is_terminal=False)] 

Beachten Sie, dass jetzt die __lt__ Methode nie genannt wird, weil die Sortierschlüssel stattdessen verwendet wird.