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

  • kgkeklikci
     + 0 comments
    import math

class Complex(object):
    def __init__(self, real, imaginary):
        self.real = real
        self.imaginary = imaginary
        self.c = complex(*(real, imaginary))

    @staticmethod
    def to_complex(no):
        return complex(*(no.real, no.imaginary))
        
    def __add__(self, no):
        c = self.c + Complex.to_complex(no)
        return Complex(c.real, c.imag)
 
    def __sub__(self, no):
        c = self.c - Complex.to_complex(no)
        return Complex(c.real, c.imag)

    def __mul__(self, no):
        c = self.c * Complex.to_complex(no)
        return Complex(c.real, c.imag)

    def __truediv__(self, no):
        c = self.c / Complex.to_complex(no)
        return Complex(c.real, c.imag)

    def mod(self):
        c = math.sqrt(self.c.real ** 2 + self.c.imag ** 2)
        return Complex(c.real, c.imag)