CODEWITHSUNDEEP
classpython-3.xoperator-overloading
srhoades28
srhoades28

Reputation: 125

Python 3 Operator Overloading

I'm trying to define the operator type add when it comes to my class Point. Point is exactly what it seems, (x, y). I can't seem to get the operator to work though because the code keeps printing the <main.Point...>. I'm pretty new to this stuff, so can someone explain what I am doing wrong? Thanks. Here is my code:

class Point:
def __init__(self, x=0, y=0):
    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,6)
p3 = p1 + p2
print(p3)

Upvotes: 0

Views: 415

Answers (1)

Kevin
Kevin

Reputation: 76254

Your add function is working as intended. It's your print that's the problem. You're getting an ugly result like <__main__.Point object at 0x027FA5B0> because you haven't told the class how you want it to display itself. Implement __str__ or __repr__ so that it shows a nice string.

class Point:
    def __init__(self, x=0, y=0):
        self.x = x
        self.y = y
    def __add__(self, other):
        return Point(self.x + other.x, self.y + other.y)
    def __repr__(self):
        return "Point({}, {})".format(self.x, self.y)
p1 = Point(3,4)
p2 = Point(5,6)
p3 = p1 + p2
print(p3)

Result:

Point(8, 10)

Upvotes: 1

Related Questions