ComplexNumber/ComplexNumber.py at master · gyandhanee/ComplexNumber · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
from __future__ import division
from math import sqrt
from math import acos, cos, sin, degrees

class ComplexNumber(object):
    def __init__(self, real, img):
        try:
            real = float(real)
            img = float(img)
        except ValueError:
            print "Enter numbers. Please."
        self.real = real
        self.img = img

    def conjugate(self):
        return ComplexNumber(self.real, -self.img)

    def __repr__(self):
        if self.img < 0:
            return "%s - %si" % (self.real, -self.img)
        elif self.img == 0:
            return "%s" % (self.real)
        else:
            return "%s + %si" % (self.real, self.img)

    def __add__(self, other):
        realpart = self.real + other.real
        imgpart = self.img + other.img
        return ComplexNumber(realpart, imgpart)

    def __sub__(self, other):
        realpart = self.real - other.real
        imgpart = self.img - other.img
        return ComplexNumber(realpart, imgpart)

    def __mul__(self, other):
        realpart = (self.real * other.real) - (self.img * other.img)
        imgpart = (self.img * other.real) + (other.img * self.real)
        return ComplexNumber(realpart, imgpart)

    def abs(self):
        return sqrt(self.real**2 + self.img**2)

    def __truediv__(self, other):
        numerator = self*other.conjugate()
        denominator = (other.abs())**2
        realpart = numerator.real/denominator
        imgpart = numerator.img/denominator
        return ComplexNumber(realpart, imgpart)

    def __pow__(self, num):
        for i in range(num):
            self = self*self
        return self

    def angle(self):
        angle_in_radians = acos(self.real/self.abs())
        angle_in_degrees = degrees(angle_in_radians)
        return angle_in_degrees

    def polarform(self):