[Python] 연산자 오버로딩, 비교 연산자

직접 정의한 클래스의 객체에

 

•  +, - * 와 같은 일반 연산자를 적용하려면 , 객체를 연산 가능한 상태로 만들어야 함

 

• 연산자 오버로딩을 통해 이를 구현

 

 

연산자 오버로딩

 

• 인스턴스 객체끼리 서로 연산을 할 수 있게 기존에 있는 연산자 의 기능을 바꾸어 중복으로 정의하는 것

 

파이썬에서는 특정 이름의 메소드를 재정의하면 연산 자 중복정의 구현

 

 

 

예제 1) 

#연산자 오버로딩 예제 1
class NumBox:
    def __init__(self,num):
        self.num = num

    def __add__(self, num):
        self.num += num

    def __sub__(self, num):
        self.num -= num

n = NumBox(40)
n + 100 # n+100 == n.__add__
print(n.num)

n-100
print(n.num)

 

예제 2) 

#연산자 오버로딩 2
#self : 객체 자신
# other : 더하고자 하는 값

class Number:
    def __init__(self,n):
        self.n = n
n1 = Number(3)
n2 = Number(4)

### 위는 에러나는 전코드, 아래는 변경 코드 ###
class Number:
    def __init__(self,n):
        self.n = n

    def __add__(self, other):
        if isinstance(other, Number):
            return Number(self.n +other.n)
        elif isinstance(other, int):
            return Number(self.n + other)


n1 = Number(3)
n2 = Number(4)

print(n1+n2)
# <__main__.Number object at 0x03248040>
print((n1+n2).n)

print(n1 + 3)
# <__main__.Number object at 0x03248058>
print((n1 + 3).n)

 

비교 연산자 오버로딩 메소드

 

비교 연산자 오버로딩 예시)

class Number:
    def __init__(self,n):
        self.n = n

    def __lt__(self, other):
        return self.n < other.n

    def __le__(self, other):
        return self.n <= other.n

    def __gt__(self, other):
        return self.n > other.n

    def __ge__(self, other):
        return self.n <= other.n

    def __eq__(self, other):
        return self.n == other.n
n1 = Number(3)
n2 = Number(4)

print('n1 < n2 : {}'.format(n1<n2))
print('n1 <= n2 : {}'.format(n1<=n2))
print('n1 > n2 : {}'.format(n1>n2))
print('n1 <= n2 : {}'.format(n1<=n2))
print('n1 == n2 : {}'.format(n1==n2))