파이썬에서 복소수에 대한 클래스를 생성하는 방법은 무엇입니까?

Question

먼저, scipy에 복소수에 대한 데이터 구조가 있다는 것을 알고 있습니다. 저는 전산 물리학 교과서를 단지 따르고 ​​있습니다. 그리고 이것은 하나의 질문입니다.

class complex: 
    def __init__(self,x,y): 
     self.re=x 
     self.im=y 
    def __add__(self, other): 
     return complex(self.re+other.re, self.im+other.im) 
    def __sub__(self, other): 
     return complex(self.re-other.re, self.im-other.im)   
    def __mul__(self, other): 
     return complex(self.re*other.re - self.im*other.im, 
     self.re*other.im + self.im*other.re)   
    def __repr__(self): 
     return '(%f , %f)' %(self.re, self.im) 

을하지만 어떻게 내가이 생성 한 클래스 부문, 복소수, 계수 및 위상을 구현해야합니다 : 는 지금까지 내가있어 무엇인가?

감사

Answer 1

부문 당신은 __abs__와 __div__ 및 계수로 구현해야합니다. 복잡한 접합과 단계에서는 고유 한 메소드 이름을 선택해야합니다. 예를 들어,

def conj(self): 
    return complex(self.re, - self.im) 

(z.conj()). 클래스에 새로운 파이썬 구문을 정의 할 수 없으므로, 예를 들어 z*을 사용할 수 없습니다. 원한다면 오른쪽 곱셈을 __rmul__으로 정의하고 __pow__을 정의해야합니다. 단항 마이너스, __neg__을 잊지 마십시오. 그러나 다른 이중 밑줄 연산자 메서드가 모두 복소수로 구현하는 것이 합리적이지는 않습니다. a list in the docs이 있습니다.