【Python 运算符重载】Python 运算符重载
class MyComplex:
def __init__(self, _real, _imag):
self.real = _real
self.imag = _imag
def __add__(self, comp):
if type(self) is type(comp):
return MyComplex(self.real+comp.real, self.imag+comp.imag)
else:
return MyComplex(self.real+comp, self.imag)
def __sub__(self, comp):
if type(self) is type(comp):
return MyComplex(self.real-comp.real, self.imag-comp.imag)
else:
return MyComplex(self.real-comp, self.imag)
def __mul__(self, comp):
if type(self) is type(comp):
return MyComplex(self.real*comp.real-self.imag*comp.imag,
self.real*comp.imag+self.imag*comp.real)
else:
return MyComplex(self.real*comp, self.imag*comp)
def __str__(self):
if self.imag>=0:
return f'{self.real}+{self.imag}i'
else:
return f'{self.real}{self.imag}i'def main():
c1 = MyComplex(3,-5)
c2 = MyComplex(2,3)
print(c1,c2)
print('c1+c2=', c1+c2)
print('c1-c2=', c1-c2)
print('c1*c2=', c1*c2)
if __name__=="__main__":
main()
