Python NumPy emath.sqrt()로 입력값의 제곱근 계산하기

Python NumPy에서 np.emath.sqrt() 메서드를 사용하면 입력값의 제곱근을 간편하게 계산할 수 있습니다. 이 메서드는 x의 제곱근을 반환하며, 입력이 스칼라라면 스칼라를, 배열이라면 배열을 결과로 돌려줍니다. 매개변수 x는 입력값을 의미하며, 특히 음수 입력 요소에 대해서는 복소수 값을 반환한다는 점이 일반적인 sqrt() 함수와 다른 큰 장점입니다.

단계별 진행 과정

먼저 필요한 라이브러리를 임포트합니다.

import numpy as np

array() 메서드를 사용해 NumPy 배열을 생성합니다.

arr = np.array([1, 4, 9, 16, 25, 36])

생성한 배열을 화면에 출력합니다.

print("Our Array...\n",arr)

배열의 차원(dimension)을 확인합니다.

print("\nDimensions of our Array...\n",arr.ndim)

배열 객체의 데이터 타입(datatype)을 확인합니다.

print("\nDatatype of our Array object...\n",arr.dtype)

배열 객체의 형태(shape)를 확인합니다.

print("\nShape of our Array object...\n",arr.shape)

입력값의 제곱근을 계산하려면 Python NumPy의 scimath.sqrt() 메서드를 사용합니다. 이 메서드는 x의 제곱근을 반환하며, 입력이 스칼라면 스칼라가, 그렇지 않으면 배열이 반환됩니다.

print("\nResult...\n",np.emath.sqrt(arr))

전체 예제 코드

import numpy as np

# array() 메서드를 사용해 NumPy 배열 생성
arr = np.array([1, 4, 9, 16, 25, 36])

# 배열 출력
print("Our Array...\n",arr)

# 차원 확인
print("\nDimensions of our Array...\n",arr.ndim)

# 데이터 타입 확인
print("\nDatatype of our Array object...\n",arr.dtype)

# 형태(shape) 확인
print("\nShape of our Array object...\n",arr.shape)

# scimath.sqrt() 메서드로 입력값의 제곱근 계산
# 입력이 스칼라면 스칼라, 배열이면 배열이 반환됩니다.
print("\nResult...\n",np.emath.sqrt(arr))

실행 결과

Our Array...
[ 1 4 9 16 25 36]

Dimensions of our Array...
1

Datatype of our Array object...
int64

Shape of our Array object...
(6,)

Result...
[1. 2. 3. 4. 5. 6.]

위 실행 결과에서 볼 수 있듯이, 정수형 배열 [1, 4, 9, 16, 25, 36]의 각 요소에 대해 제곱근 [1.0, 2.0, 3.0, 4.0, 5.0, 6.0]이 부동소수점 배열로 반환되었습니다. 만약 배열에 음수 값이 포함되어 있다면 오류 없이 복소수 형태의 결과가 자동으로 반환되므로, 음수까지 고려한 수학적 연산에 유용하게 활용할 수 있습니다.