build.py

# %load q01_create_class/build.py
import pandas as pd
import numpy as np
import math

'write your solution here'


class complex_number:
    '''The complex number class.

    Attributes:
    attr1 (x): Real part of complex number.
    attr2 (y): Imaginary part of complex number.

    '''
    def __init__(self,real,imag):
        self.real = real
        self.imag = imag
    
    def __add__(self, other):
        x=self.real+other.real
        y=self.imag+other.imag
        return complex_number(x,y)
    
    def __sub__(self,other):
        x=self.real-other.real
        y=self.imag-other.imag
        return complex_number(x,y)
    
    def __mul__(self,other):
        x=(self.real*other.real)-(self.imag*other.imag)
        y= (other.real*self.imag)+(self.real*other.imag)
        return complex_number(x,y)
    
    def __truediv__(self,other):
        x=((self.real*other.real)+(self.imag*other.imag))/((other.real**2+ other.imag**2))
        y=((self.imag*other.real)-(self.real*other.imag))/((other.real**2+ other.imag**2))
        return x,y
   
    def __str__(self):
        if '-' in str(self.imag):
            return '{0}-i{1}'.format(self.real,-self.imag)
        else :
            return '{0}+i{1}'.format(self.real,self.imag)
    
    def abs(self):
        return (math.sqrt(float(self.real)**2+float(self.imag)**2))

    def conjugate(self):
        return complex_number((self.real),(self.imag*-1))
    
    def argument(self):
        return math.degrees(math.atan(self.imag/self.real))