Python NumPy – 2D 계수 배열로 점 (x, y, z)에서 3D 에르미트 급수 평가하기

Python NumPy에서 hermite.hermval3d() 메서드를 사용하면 점 (x, y, z)에서 3차원 에르미트(Hermite) 급수를 손쉽게 평가할 수 있습니다. 에르미트 다항식은 물리학, 특히 양자역학과 확률론에서 널리 활용되는 직교 다항식으로, NumPy의 polynomial 모듈은 이를 다루기 위한 다양한 도구를 제공합니다. 이 메서드는 x, y, z에서 대응하는 값들의 조합으로 구성된 점들에 대해 다차원 다항식의 값을 반환합니다.

hermval3d() 메서드의 주요 매개변수

첫 번째 매개변수는 x, y, z입니다. 3차원 급수는 점 (x, y, z)에서 평가되며, x, y, z는 서로 같은 형태(shape)를 가져야 합니다. 만약 x, y, z 중 하나라도 리스트나 튜플로 전달되면 먼저 ndarray로 변환되며, ndarray가 아닌 값은 스칼라로 취급됩니다.

두 번째 매개변수인 C는 계수 배열입니다. 다중 차수(multidegree) i, j, k에 해당하는 항의 계수가 c[i, j, k]에 저장되도록 정렬되어 있습니다. c의 차원이 3보다 큰 경우, 나머지 인덱스들은 여러 개의 계수 집합을 나타내는 데 사용됩니다.

단계별 구현 방법

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

import numpy as np
from numpy.polynomial import hermite as H

계수로 사용할 2D 배열을 생성합니다 −

c = np.arange(4).reshape(2,2)

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

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

배열의 차원을 확인합니다 −

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

배열의 데이터 타입을 확인합니다 −

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

배열의 형태(shape)를 확인합니다 −

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

점 (x, y, z)에서 3D 에르미트 급수를 평가하려면 Python NumPy의 hermite.hermval3d() 메서드를 사용합니다 −

print("\nResult...\n",H.hermval3d([1,2],[1,2],[1,2],c))

전체 예제 코드

import numpy as np
from numpy.polynomial import hermite as H

# 계수로 사용할 2D 배열 생성
c = np.arange(4).reshape(2,2)

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

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

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

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

# hermite.hermval3d() 메서드로 점 (x, y, z)에서 3D 에르미트 급수 평가
print("\nResult...\n",H.hermval3d([1,2],[1,2],[1,2],c))

실행 결과

Our Array...
    [[0 1]
    [2 3]]

Dimensions of our Array...
2

Datatype of our Array object...
int64

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

Result...
    [138. 258.]

위 실행 결과에서 볼 수 있듯이, 2×2 크기의 계수 배열을 사용하여 점 (1, 1, 1)과 (2, 2, 2)에서 각각 138.0과 258.0이라는 3D 에르미트 급수의 평가값을 얻을 수 있습니다. 이처럼 hermval3d() 메서드를 활용하면 복잡한 다차원 다항식 연산을 간결하고 효율적으로 처리할 수 있습니다.