Custom Comparator for Objects - Heapq

heapq custom comparator.py

import heapq
from collections import Counter


class Obj:
    def __init__(self, a, b):
        self.a = a
        self.b = b

    def __lt__(self, other):
        if self.a < other.a:
            return True
        elif self.a > other.a:
            return False
        elif self.b < other.b:
            return True
        elif self.b > other.b:
            return False

    def __gt__(self, other):
        if self.a > other.a:
            return True
        elif self.a > other.a:
            return False
        elif self.b > other.b:
            return True
        elif self.b < other.b:
            return False

    def __eq__(self, other):
        return self.a == other.b and self.b == other.b


nums = [1, 1, 2, 2, 111]

lst = []

mp = Counter(nums)

for k, v in mp.items():
    heapq.heappush(lst, Obj(v, k))

while lst:
    print(heapq.heappop(lst).__dict__)