Přetěžování operátorů: porovnávání objektů

Úvod do objektově orientovaného programování v Pythonu

George Boorman

Curriculum Manager, DataCamp

Rovnost objektů

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
Úvod do objektově orientovaného programování v Pythonu

Rovnost objektů

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
Úvod do objektově orientovaného programování v Pythonu

Proměnné jsou reference

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>
  • Výstup print() odkazuje na místo v paměti, kam je proměnná uložena
  • == porovnává reference, nikoli data
Úvod do objektově orientovaného programování v Pythonu

Vlastní porovnání

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

list_one == list_two
True
Úvod do objektově orientovaného programování v Pythonu

Metoda __eq__()

  • __eq__() se volá při porovnání dvou objektů třídy pomocí ==
  • Přijímá 2 argumenty, self a other – porovnávané objekty
  • Vrací hodnotu Boolean
Úvod do objektově orientovaného programování v Pythonu

Metoda __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)
Úvod do objektově orientovaného programování v Pythonu

Porovnání objektů

# 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
Úvod do objektově orientovaného programování v Pythonu

Kontrola typů

  • Co když mají dva objekty různých tříd stejné atributy a hodnoty?
    • Python je vyhodnotí jako rovné
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))
Úvod do objektově orientovaného programování v Pythonu

Další operátory porovnání

Operátor Metoda
== __eq__()
!= __ne__()
>= __ge__()
<= __le__()
> __gt__()
< __lt__()
  • Přizpůsobte definicí uvnitř třídy
Úvod do objektově orientovaného programování v Pythonu

Vyzkoušejte si to!

Úvod do objektově orientovaného programování v Pythonu

Preparing Video For Download...