Overloading Arithmetic Operators with add mul

Learn how to overload arithmetic operators in Python using __add__ and __mul__ to make custom classes behave like built-in types.

Overloading Arithmetic Operators in Python with __add__ and __mul__

In Python, operator overloading allows you to define how operators like + and * behave for your custom objects. This is done by implementing special methods like __add__ for addition and __mul__ for multiplication. This feature lets your classes integrate naturally with Python syntax, improving code readability and usability.

Why Overload Operators?

Overloading operators makes custom classes behave like native types. For instance, a Vector class can define addition and multiplication in a way that supports v1 + v2 or v1 * 3.

Implementing __add__ and __mul__

To overload arithmetic operators, define __add__(self, other) for addition and __mul__(self, other) for multiplication:

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

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

    def __mul__(self, scalar):
        return Vector(self.x * scalar, self.y * scalar)

    def __repr__(self):
        return f\\"Vector(self.x, self.y)\\"

Using the Overloaded Operators

Create vectors and use + and * as you would with numbers:

v1 = Vector(2, 3)
v2 = Vector(4, 5)
v3 = v1 + v2  # Vector addition
v4 = v1 * 3   # Scalar multiplication

print(v3)  # Output: Vector(6, 8)
print(v4)  # Output: Vector(6, 9)

Best Practices

• Ensure __add__ checks types of other to avoid errors.
• Support reversed operations with __radd__ and __rmul__ if needed.
• Keep methods simple and consistent with Python’s expected behavior.

Conclusion

Overloading arithmetic operators makes custom classes more intuitive and Pythonic. With __add__ and __mul__, your objects can participate in natural arithmetic expressions, enabling cleaner and more expressive code.

Subscribe to Our YouTube for More

Download as PDF

https://blog.arashtad.com/updates/overloading-arithmetic-operators-with-add-mul/?feed_id=12358&_unique_id=688488beac96a