Python NumPy로 복소수 근을 이용해 Hermite_e 시리즈 생성하기

Python의 NumPy 라이브러리에서 주어진 복소수 근(roots)을 기반으로 Hermite_e 시리즈를 생성하려면 hermite_e.hermefromroots() 메서드를 사용하면 됩니다.

이 메서드는 계수(coefficients)들로 이루어진 1차원 배열을 반환합니다. 모든 근이 실수라면 결과 배열도 실수형이 되지만, 근 중 하나라도 복소수가 포함되어 있다면 최종 계수가 모두 실수 값이더라도 결과 배열은 복소수형(complex)으로 반환됩니다. 매개변수 roots에는 근들을 담고 있는 시퀀스(sequence)를 전달합니다.

진행 단계

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

from numpy.polynomial import hermite_e as H

주어진 복소수 근을 사용하여 Hermite_e 시리즈를 생성합니다.

j = complex(0,1)
print("Result...\n",H.hermefromroots((-j, j)))

결과 배열의 데이터 타입(dtype)을 확인합니다.

print("\nType...\n",H.hermefromroots((-j, j)).dtype)

결과 배열의 형태(shape)를 확인합니다.

print("\nShape...\n",H.hermefromroots((-j, j)).shape)

전체 예제 코드

from numpy.polynomial import hermite_e as H

j = complex(0,1)
print("Result...\n",H.hermefromroots((-j, j)))

# 데이터 타입 확인
print("\nType...\n",H.hermefromroots((-j, j)).dtype)

# 형태(shape) 확인
print("\nShape...\n",H.hermefromroots((-j, j)).shape)

실행 결과

Result...
    [2.+0.j 0.+0.j 1.+0.j]

Type...
complex128

Shape...
(3,)

위 실행 결과에서 볼 수 있듯이, 허수 단위 j와 -j를 근으로 전달했기 때문에 결과 배열은 실수 값을 포함하고 있음에도 불구하고 complex128 복소수 타입으로 반환됩니다. 또한 두 개의 근으로부터 3개의 계수가 생성되어 배열의 shape은 (3,)이 됩니다.