Next
1. Create Class & Object2. Instance vs Class Variables3. Constructors in Python Classes4. Single Inheritance5. Multiple Inheritance6. Multilevel Inheritance7. Polymorphism with Methods8. Method Overloading Simulation9. Encapsulation with Getters/Setters10. Abstraction with ABC11. Operator Overloading12. Custom Iterator13. Class Methods14. Static Methods15. Composition Example16. Aggregation Example17. Private Variables18. Magic Methods Overview19. Object Cloning20. Custom Exceptions21. Class Decorator22. Prototype Pattern23. Singleton Pattern24. MRO (Method Resolution Order) Demo25. Polymorphism (Duck Typing)

Operator Overloading in Python

Overload the + operator for a custom class using __add__.

IntermediateObject-Oriented ProgramsExample 11 of 25
operator-overloading.py
Run in browser
1# Program to demonstrate operator overloading
2

3class Vector:
4    def __init__(self, x, y):
5        self.x = x
6        self.y = y
7

8    def __add__(self, other):
9        return Vector(self.x + other.x, self.y + other.y)
10

11    def __repr__(self):
12        return f"Vector({self.x}, {self.y})"
13

14v1 = Vector(1, 2)
15v2 = Vector(3, 4)
16print(v1 + v2)

Output

Vector(4, 6)

What's going on

Implementing __add__ allows the + operator to work naturally with Vector instances.

PreviousAbstraction with ABCNextCustom Iterator