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):
        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
        imaginary =self.real*no.imaginary+ self.imaginary*no.real
        return Complex(real, imaginary)
    
    def __truediv__(self, no):
        denominator = float(no.real**2 +no.imaginary**2)
        Numerator = self*Complex(no.real,-no.imaginary)
        real =  Numerator.real/denominator
        imaginary = Numerator.imaginary/denominator
        return Complex(real, imaginary)
    
    def mod(self):
        real = math.sqrt(self.real**2 +self.imaginary**2)
        return Complex(real,0)
    
    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
    

    if name == 'main': c = map(float, input().split()) d = map(float, input().split()) x = Complex(*c) y = Complex(*d) print(*map(str, [x+y, x-y, x*y, x/y, x.mod(), y.mod()]), sep='\n')