Commit Live PR

q01_create_class/build.py

@@ -1,15 +1,57 @@
-import pandas as pd
-import numpy as np
 import math
+import numpy as np
+
+class complex_number():
+
+    def __init__(self, real=0.0,imag=0.0):
+        self.real=real #real part
+        self.imag=imag #imaginary part
+
+    def __repr__(self):
+        if self.real == 0.0 and self.imag == 0.0:
+            return '0.00'
+        if self.real == 0:
+            return '%.2fi' % self.imag
+        if self.imag == 0:
+            return '%.2f' % self.real
+        return '%.2f %s %.2fi' % (self.real, '+' if self.imag >= 0 else '-', abs(self.imag))
+
+    def abs(self):
+        return (self.real**2+self.imag**2)**0.5
+
+    def argument(self):
+        return math.degrees(np.arctan(self.real/self.imag))
+
+    def conjugate(self):
+        return complex_number(self.real,-self.imag)
+
+    def __add__(self,other):
+        a1=self.real+other.real
+        b1=self.imag+other.imag
+        return complex_number(a1,b1)
+
+    def __sub__(self,other):
+        a1=self.real-other.real
+        b1=self.imag-other.imag
+        return complex_number(a1,b1)
+
+    def __mul__(self,other):
+        a1= (self.real*other.real)-(self.imag*other.imag)
+        b1= (self.real*other.imag)+(other.real*self.imag)
+        return complex_number(a1,b1)
+
+    def __truediv__(self,other):
+        c= ((self.real*other.real+self.imag*other.imag))/(other.real**2+other.imag**2)
+        d= ((self.imag*other.real-self.real*other.imag))/(other.real**2+other.imag**2)
+        return c,d
 
-"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.
 
-    """