Перегрузка операторов: сравнение объектов

Введение в объектно-ориентированное программирование на Python

George Boorman

Curriculum Manager, DataCamp

Равенство объектов

class Customer:
    def __init__(self, name, balance):
        self.name, self.balance = name, balance  

customer1 = Customer("Maryam Azar", 3000)
customer2 = Customer("Maryam Azar", 3000)


# Check for equality
customer1 == customer2

False
Введение в объектно-ориентированное программирование на Python

Равенство объектов

class Customer:
    def __init__(self, name, balance, acc_id):
        self.name, self.balance = name, balance
        self.acc_id = acc_id    

customer1 = Customer("Maryam Azar", 3000, 123)
customer2 = Customer("Maryam Azar", 3000, 123)

customer1 == customer2

False
Введение в объектно-ориентированное программирование на Python

Переменные — это ссылки

customer_one = Customer("Maryam Azar", 3000, 123)
customer_two = Customer("Maryam Azar", 3000, 123)

print(customer_one)

<__main__.Customer at 0x1f8598e2e48>

print(customer_two)

<__main__.Customer at 0x1f8598e2240>
  • Вывод print() указывает на область памяти, в которой хранится переменная
  • == сравнивает ссылки, а не данные
Введение в объектно-ориентированное программирование на Python

Пользовательское сравнение

# Two different lists containing the same data
list_one = [1,2,3]
list_two = [1,2,3]

list_one == list_two

True
Введение в объектно-ориентированное программирование на Python

Метод __eq__()

  • __eq__() вызывается при сравнении двух объектов класса с помощью ==
  • Принимает 2 аргумента: self и other — сравниваемые объекты
  • Возвращает булево значение
Введение в объектно-ориентированное программирование на Python

Метод __eq__()

class Customer:
    def __init__(self, acc_id, name):
        self.acc_id, self.name = acc_id, name

    # Will be called when == is used
    def __eq__(self, other):

        # Printout
        print("__eq__() is called")

        # Returns True if all attributes match
        return (self.acc_id == other.acc_id) and (self.name == other.name)
Введение в объектно-ориентированное программирование на Python

Сравнение объектов

# Two equal objects
customer1 = Customer(123, "Maryam Azar")
customer2 = Customer(123, "Maryam Azar")

customer1 == customer2

__eq__() is called
True

# Two unequal objects - different ids
customer1 = Customer(123, "Maryam Azar")
customer2 = Customer(456, "Maryam Azar")

customer1 == customer2

__eq__() is called
False
Введение в объектно-ориентированное программирование на Python

Проверка типов

  • Что если объекты разных классов имеют одинаковые атрибуты и значения?
    • Python посчитает их равными
class Customer:
    def __init__(self, acc_id, name):
        self.acc_id, self.name = idacc_id name

    def __eq__(self, other):
        # Returns True if the objects have the same attributes
        # and are of the same type
        return (self.acc_id == other.acc_id) and (self.name == other.name)\
            and (type(self) == type(other))
Введение в объектно-ориентированное программирование на Python

Другие операторы сравнения

Оператор Метод
== __eq__()
!= __ne__()
>= __ge__()
<= __le__()
> __gt__()
< __lt__()
  • Настройте поведение, определив методы в классе
Введение в объектно-ориентированное программирование на Python

Давайте потренируемся!

Введение в объектно-ориентированное программирование на Python

Preparing Video For Download...