Skip to main content

Operator Overloading in Python

 

Learning Sections          show

Operator Overloading in Python

Operator overloading in Python allows you to define custom behavior for operators when they are used with user-defined classes. This makes your objects behave more like built-in types, which can lead to more intuitive and readable code. Operator overloading is implemented by defining special methods, also known as magic methods or dunder methods (because they begin and end with double underscores).

For example, to overload the addition operator +, you need to define the __add__ method in your class.

Example: Overloading the Addition Operator

class Point:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __add__(self, other):
        return Point(self.x + other.x, self.y + other.y)

    def __repr__(self):
        return f"Point({self.x}, {self.y})"

p1 = Point(1, 2)
p2 = Point(3, 4)
p3 = p1 + p2

print(p3)  # Output: Point(4, 6)
Explanation:

In the example above, the __add__ method is defined to overload the addition operator +. When we create two instances of the Point class and add them using p1 + p2, Python calls the __add__ method, which returns a new Point instance with the coordinates summed.

Other Common Operators

Here are some other common operators and their corresponding magic methods:

  1. + : __add__(self, other)
  2. - : __sub__(self, other)
  3. * : __mul__(self, other)
  4. / : __truediv__(self, other)
  5. // : __floordiv__(self, other)
  6. % : __mod__(self, other)
  7. ** : __pow__(self, other)
  8. == : __eq__(self, other)
  9. != : __ne__(self, other)
  10. > : __gt__(self, other)
  11. < : __lt__(self, other)
  12. >= : __ge__(self, other)
  13. <= : __le__(self, other)
Example: Overloading Comparison Operators

class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age

    def __eq__(self, other):
        return self.age == other.age

    def __lt__(self, other):
        return self.age < other.age

person1 = Person("Alice", 30)
person2 = Person("Bob", 25)
person3 = Person("Charlie", 30)

print(person1 == person3)  # Output: True
print(person1 < person2)   # Output: False
Explanation:

In this example, we overload the equality operator == using the __eq__ method and the less than operator < using the __lt__ method. This allows us to compare Person objects based on their age attribute.

Labels:

Popular posts from this blog

Generators in Python

  Learning Sections          show Generators in Python Generators are a special type of iterator in Python that allow you to iterate over a sequence of items without storing them all in memory at once. They are useful for generating large sequences of data on-the-fly, or for processing data in a memory-efficient manner. Creating Generators In Python, generators are created using generator functions or generator expressions: # Generator function def my_generator ( n ): for i in range ( n ): yield i # Generator expression my_generator = ( i for i in range ( 10 )) A generator function uses the yield keyword to yield values one at a time, while a generator expression creates an anonymous generator. Iterating Over Generators You can iterate over the values produced by a generator using a for loop: for value in my_generator ( 5 ): print ( value ) This w...
Read more

Inheritance in Python

  Learning Sections          show Inheritance in Python Inheritance is a fundamental concept in object-oriented programming (OOP) that allows a class to inherit attributes and methods from another class. The class that inherits is called the child class or subclass, and the class being inherited from is called the parent class or superclass. Basic Inheritance In Python, a child class inherits from a parent class by specifying the parent class in parentheses after the child class name. Example: class Animal : def __init__ ( self , name ): self . name = name def speak ( self ): raise NotImplementedError ( "Subclass must implement this method" ) class Dog ( Animal ): def speak ( self ): return "Woof!" class Cat ( Animal ): def speak ( self ): return "Meow!" # Create instances of Dog and Cat dog = Dog ( "Buddy" ) cat = Cat ( "Whiskers" ...
Read more

If else Conditional Statements in Python

  Learning Sections     show If-Else Conditional Statements Conditional statements allow you to execute different blocks of code based on certain conditions. The most common conditional statement is the if statement. It can be used alone, or combined with elif (else if) and else statements to handle multiple conditions. If Statement The if statement evaluates a condition, and if the condition is true, the block of code indented under the if statement is executed. # If statement example x = 10 if x >> 0 : print ( "x is positive" ) If-Else Statement The if-else statement adds an additional block of code that runs if the condition is false. # If-else statement example x = -10 if x >> 0 : print ( "x is positive" ) else : print ( "x is non-positive" ) If-Elif-Else Statement The if-elif-else statement allows you to check multiple conditions. The fir...
Read more