Arithmetic operator overloading in Python - Mini Project: Build & Apply

Arithmetic Operator Overloading

📖 Scenario: Imagine you are creating a simple program to handle points on a 2D plane. You want to add two points together using the + operator, just like adding numbers.

🎯 Goal: You will create a class called Point that stores x and y coordinates. Then, you will make the + operator work to add two points by adding their x and y values.

📋 What You'll Learn

Create a class called Point with x and y attributes

Add a method to overload the + operator to add two Point objects

Create two Point objects with given coordinates

Add the two points using the + operator and print the result

💡 Why This Matters

🌍 Real World

Operator overloading helps make custom objects behave like built-in types, making code easier to read and write, such as adding points, vectors, or complex numbers.

💼 Career

Understanding operator overloading is useful in software development roles that involve creating custom data types, game development, graphics programming, and scientific computing.

1

Create the Point class with x and y attributes

Write a class called Point with an __init__ method that takes x and y as parameters and stores them as attributes.

Python

# Your code here

Hint

Use def __init__(self, x, y): to create the constructor and assign x and y to self.x and self.y.

2

Create two Point objects with coordinates (3, 4) and (5, 7)

Create two variables called p1 and p2 that are Point objects with coordinates (3, 4) and (5, 7) respectively.

Python

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

# Your code here

Hint

Use p1 = Point(3, 4) and p2 = Point(5, 7) to create the points.

3

Add a method to overload the + operator to add two Point objects

Inside the Point class, add a method called __add__ that takes self and other as parameters. It should return a new Point with x as the sum of self.x and other.x, and y as the sum of self.y and other.y.

Python

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

    # Your code here

p1 = Point(3, 4)
p2 = Point(5, 7)

Hint

Define def __add__(self, other): and return a new Point with summed x and y.

4

Add the two points using + and print the result

Create a variable called p3 that is the sum of p1 and p2 using the + operator. Then, print the x and y values of p3 separated by a space using print(p3.x, p3.y).

Python

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)

p1 = Point(3, 4)
p2 = Point(5, 7)

# Your code here

Hint

Use p3 = p1 + p2 and then print(p3.x, p3.y) to show the result.

Practice

1. What does arithmetic operator overloading allow you to do in Python?

easy

A. Define how operators like +, -, * work for your custom objects

B. Change the behavior of built-in data types like int and str

C. Create new arithmetic operators not available in Python

D. Automatically optimize arithmetic operations for speed

Solution

1. Step 1: Understand operator overloading concept

   Operator overloading lets you tell Python how to use operators like + or * with your own objects.

2. Step 2: Identify what can be customized

   You can define special methods like __add__ to customize + for your class instances.

3. Final Answer:

   Define how operators like +, -, * work for your custom objects -> Option A

4. Quick Check:

   Operator overloading = custom operator behavior [OK]

Hint: Think: 'How do I make + work for my objects?' [OK]

Common Mistakes:

• Confusing operator overloading with creating new operators
• Thinking it changes built-in types behavior
• Assuming it improves performance automatically

2. Which special method should you define to overload the addition operator (+) in a Python class?

easy

A. __append__

B. __plus__

C. __sum__

D. __add__

Solution

1. Step 1: Recall Python special method names for operators

   Python uses __add__ to overload the + operator in classes.

2. Step 2: Check other options

   __plus__, __sum__, and __append__ are not valid special methods for + operator.

3. Final Answer:

   __add__ -> Option D

4. Quick Check:

   + operator method = __add__ [OK]

Hint: Remember: add = __add__, subtract = __sub__ [OK]

Common Mistakes:

• Using __plus__ instead of __add__
• Confusing __sum__ with sum() function
• Using __append__ which is for lists

3. What will be the output of this code?

class Number:
    def __init__(self, value):
        self.value = value
    def __add__(self, other):
        return Number(self.value + other.value)
    def __str__(self):
        return str(self.value)

n1 = Number(5)
n2 = Number(10)
print(n1 + n2)

medium

A. TypeError

B. Number object at some memory address

C. 15

D. 5 10

Solution

1. Step 1: Understand __add__ method behavior

   Adding n1 + n2 calls __add__, which returns a new Number with value 5 + 10 = 15.

2. Step 2: Understand __str__ method effect on print

   print calls __str__ on the result, which returns '15' as string.

3. Final Answer:

   15 -> Option C

4. Quick Check:

   Custom + returns Number(15), printed as '15' [OK]

Hint: Check __add__ return and __str__ print output [OK]

Common Mistakes:

• Expecting print to show object memory address
• Forgetting __str__ method for printing
• Thinking it raises TypeError without __add__

4. Find the error in this code that tries to overload the multiplication operator (*) for a class:

class Multiplier:
    def __init__(self, num):
        self.num = num
    def __mul__(self, other):
        return self.num * other.num

m1 = Multiplier(3)
m2 = Multiplier(4)
print(m1 * m2)

medium

A. The __mul__ method should return a Multiplier object, not an int

B. The __init__ method is missing a return statement

C. The print statement should use str(m1 * m2)

D. The __mul__ method should be named __multiply__

Solution

1. Step 1: Check __mul__ return type

   __mul__ returns an int (self.num * other.num), but operator overloading usually returns an object of the class.

2. Step 2: Understand why returning int is a problem

   Returning int means further chained operations on Multiplier objects will fail or behave unexpectedly.

3. Final Answer:

   The __mul__ method should return a Multiplier object, not an int -> Option A

4. Quick Check:

   __mul__ must return class instance for chaining [OK]

Hint: Return class instance in operator methods, not raw values [OK]

Common Mistakes:

• Returning raw int instead of class instance
• Thinking __init__ needs return
• Misnaming __mul__ method

5. You want to create a class Vector that supports adding vectors and multiplying by a number. Which methods should you define to support v1 + v2 and v1 * 3 where v1 and v2 are Vector objects?

hard

A. __add__ for vector + vector, __rmul__ for number * vector

B. __add__ for vector + vector, __mul__ for vector * number

C. __add__ for vector + vector, __mul__ for number * vector

D. __radd__ for vector + vector, __mul__ for vector * number

Solution

1. Step 1: Identify method for vector + vector

   __add__ handles adding two Vector objects like v1 + v2.

2. Step 2: Identify method for vector * number

   __mul__ handles multiplying Vector by a number like v1 * 3.

3. Final Answer:

   __add__ for vector + vector, __mul__ for vector * number -> Option B

4. Quick Check:

   Use __add__ and __mul__ for these operations [OK]

Hint: Use __add__ for +, __mul__ for * with your class [OK]

Common Mistakes:

• Confusing __rmul__ with __mul__
• Using __radd__ instead of __add__ for vector + vector
• Assuming number * vector uses __mul__ (it uses __rmul__)