Python에서 복소수의 제곱근을 구하는 방법

Python에서 복소수(complex number)의 제곱근을 구하려면 cmath 라이브러리를 사용하면 됩니다. cmath는 복소수 연산을 전문적으로 처리하기 위해 제공되는 표준 라이브러리로, 일반적인 math 모듈과 달리 음수나 복소수 입력에 대해서도 오류 없이 결과를 반환합니다.

cmath 모듈의 sqrt() 함수를 사용하면 복소수의 제곱근을 간단하게 계산할 수 있습니다. 아래 예제를 살펴보겠습니다.

예제

from cmath import sqrt

a = 0.2 + 0.5j
print(sqrt(a))

출력 결과

위 코드를 실행하면 다음과 같은 출력을 얻을 수 있습니다.

(0.6076662244659689+0.4114100635092987j)

참고 사항

만약 일반적인 math.sqrt() 함수를 사용하면, 입력값이 복소수일 경우 ValueError가 발생합니다. 따라서 복소수를 다룰 때는 반드시 cmath 모듈을 사용해야 합니다.

또한 cmath.sqrt()는 항상 복소수 형태의 결과를 반환합니다. 실수만 필요한 경우에는 .real 속성으로 실수부를 추출할 수 있습니다.