Classes: Dealing with Complex Numbers

  • + 0 comments
    import math
    
    
    class Complex(object):
        def __init__(self, real, imaginary):
            self.real = real
            self.imaginary = imaginary
    
        def __add__(self, no):
            if isinstance(no, Complex):
                real = self.real + no.real
                imaginary = self.imaginary + no.imaginary
                return str(self.__class__(real, imaginary))
    
        def __sub__(self, no):
            if isinstance(no, Complex):
                real = self.real - no.real
                imaginary = self.imaginary - no.imaginary
                return str(self.__class__(real, imaginary))
    
        def __mul__(self, no):
            real = (self.real * no.real) - (self.imaginary * no.imaginary)
            imaginary = (self.real * no.imaginary) + (self.imaginary * no.real)
            return str(self.__class__(real, imaginary))
    
        def __truediv__(self, no):
    
            denominator = math.pow(no.real,2) + math.pow(no.imaginary,2)
            real = ((self.real * no.real) + (self.imaginary * no.imaginary)) / denominator
            imaginary = ((self.imaginary * no.real) - (self.real * no.imaginary)) / denominator
            return str(self.__class__(real, imaginary))
    
        def mod(self):
            real = math.sqrt(math.pow(self.real,2) + math.pow(self.imaginary,2))
            imaginary = 0
            return str(self.__class__(real, imaginary))
    
    
        def __str__(self):
            if self.imaginary == 0:
                result = "%.2f+0.00i" % (self.real)
            elif self.real == 0:
                if self.imaginary >= 0:
                    result = "0.00+%.2fi" % (self.imaginary)
                else:
                    result = "0.00-%.2fi" % (abs(self.imaginary))
            elif self.imaginary > 0:
                result = "%.2f+%.2fi" % (self.real, self.imaginary)
            else:
                result = "%.2f-%.2fi" % (self.real, abs(self.imaginary))
            return result