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.

Popular posts from this blog

Introduction to OOPs in Python

  Learning Sections          show Introduction to Object-Oriented Programming (OOP) Object-Oriented Programming (OOP) is a programming paradigm that organizes software design around objects rather than actions and data rather than logic. It revolves around the concept of "objects", which are instances of classes. These objects encapsulate data, in the form of attributes or properties, and behaviors, in the form of methods or functions. OOP promotes modularity, reusability, and extensibility in software development. Key Concepts of OOP: Class: A class is a blueprint or template for creating objects. It defines the attributes (data) and methods (functions) that will characterize any object instantiated from that class. Object: An object is an instance of a class. It is a concrete realization of the class blueprint, containing actual values instead of placeholders for attributes. Encapsulation: Encapsulation is ...

Classes and Objects in Python

  Learning Sections          show Classes and Objects in Python In Python, a class is a blueprint for creating objects. An object is an instance of a class. Classes allow you to logically group data and functions in a way that is easy to manage and reuse. 1. Defining a Class To define a class in Python, you use the class keyword followed by the class name and a colon. Inside the class, you can define attributes and methods. Example: # Define a class class Person : # Class attribute species = 'Human' # Class method def greet ( self ): return 'Hello, I am a person.' # Create an object of the class person1 = Person () # Access class attribute print ( person1 . species ) # Output: Human # Call class method print ( person1 . greet ()) # Output: Hello, I am a person. 2. Creating Objects To create an object of a class, you simply call the class name followed by paren...

Exception Handling in Python

  Learning sections          show Exception Handling in Python Exception handling in Python is done through the use of try , except , else , and finally blocks. This allows you to catch and handle errors gracefully. Below are some examples and explanations: 1. Basic Try-Except The try block lets you test a block of code for errors. The except block lets you handle the error. # Example of basic try-except try : result = 10 / 0 except ZeroDivisionError : print ( "Cannot divide by zero!" ) # Output: # Cannot divide by zero! 2. Handling Multiple Exceptions You can catch multiple exceptions by specifying multiple except blocks. # Example of handling multiple exceptions try : result = 10 / 0 except ZeroDivisionError : print ( "Cannot divide by zero!" ) except TypeError : print ( "Invalid operation!" ) # Output: # Cannot divide by zero! 3. Using Else The e...