We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Python
  3. Classes
  4. Classes: Dealing with Complex Numbers
  5. Discussions

Classes: Dealing with Complex Numbers

  • paul64572
     + 0 comments

    import math

    class Complex(object):

    def result(self, d) :
    return Complex(d.real, d.imag)        

def __init__(self, real, imaginary):
    self.real = real
    self.imaginary = imaginary
    self.c = complex(real, imaginary)

def __add__(self, no):
    return self.result( self.c + complex(no.real, no.imaginary) )

def __sub__(self, no):
    return self.result( self.c - complex(no.real, no.imaginary) )

def __mul__(self, no):
    return self.result( self.c * complex(no.real, no.imaginary) )

def __truediv__(self, no):
    return self.result( self.c / complex(no.real, no.imaginary) )

def mod(self):
    return Complex(math.sqrt(self.real**2 + self.imaginary**2),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')