overload operator python code example

Example 1: overload operator python

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)

Example 2: python operator overloading

class GridPoint:
    def __init__(self, x, y):  
        self.x = x  
        self.y = y  
  
    def __add__(self, other):  	# Overloading + operator
        return GridPoint(self.x + other.x, self.y + other.y)
  
    def __str__(self):  		# Overloading "to string" (for printing)
        string = str(self.x)  
        string = string + ", " + str(self.y)  
        return string  
   	def __gt__(self, other):  	# Overloading > operator (Greater Than)
       	return self.x > other.x    
point1 = GridPoint(3, 5) 
point2 = GridPoint(-1, 4)
point3 = point1 + point2  	# Add two points using __add__() method
print(point3)  				# Print the attributes using __str__() method
if point1 > point2:  		# Compares using __gt__() method
   print('point1 is greater than point2')

Tags:

Misc Example