Classes: Dealing with Complex Numbers

  • + 0 comments
    class Complex(object):
        def __init__(self, real, imaginary):
            self.real = real
            self.imaginary = imaginary
            
        def __add__(self, no):
            return Complex(self.real + no.real, self.imaginary + no.imaginary)
            
        def __sub__(self, no):
            return Complex(self.real - no.real, self.imaginary - no.imaginary)
            
        def __mul__(self, no):
            real = self.real * no.real - self.imaginary * no.imaginary
            imag = self.real * no.imaginary + self.imaginary * no.real
            return Complex(real, imag)
    
        def __truediv__(self, no):
            denominator = no.real ** 2 + no.imaginary ** 2
            real = (self.real * no.real + self.imaginary * no.imaginary) / denominator
            imag = (self.imaginary * no.real - self.real * no.imaginary) / denominator
            return Complex(real, imag)
    
        def mod(self):
            return Complex(math.sqrt(self.real ** 2 + self.imaginary ** 2), 0)