Python에서 3차원 계수 배열을 활용해 x·y 데카르트 곱 위의 2차원 라게르 급수 평가하기

numpy.polynomial.laguerre.laggrid2d() 메서드를 사용하면 Python에서 x와 y의 데카르트 곱(Cartesian product) 위에서 2차원 라게르(Laguerre) 급수를 손쉽게 평가할 수 있습니다. 이 메서드는 x의 각 점과 y의 각 점을 조합한 모든 좌표 지점에서 2차원 라게르 급수의 값을 계산하여 반환합니다.

라게르 다항식은 물리학 분야, 특히 양자역학의 수소 원자 파동 함수나 통계 역학 문제에서 널리 활용되는 직교 다항식입니다. NumPy는 이러한 라게르 급수를 다루기 위한 다양한 도구를 제공하며, 그중 laggrid2d()는 격자(grid) 형태의 평가에 특화된 함수입니다.

laggrid2d() 메서드의 주요 특징

계수 배열 c가 2차원 미만일 경우, 자동으로 차원이 추가되어 2차원 형태로 확장됩니다. 결과 배열의 shape는 c.shape[2:] + x.shape + y.shape 공식에 따라 결정됩니다.

첫 번째 매개변수 — x, y

2차원 급수가 평가될 점들을 의미합니다. x 또는 y가 리스트(list)나 튜플(tuple)로 전달되면 먼저 ndarray로 변환되며, ndarray가 아니라면 스칼라(scalar)로 취급됩니다.

두 번째 매개변수 — c

계수(coefficient)들이 담긴 배열입니다. 다중 차수(multidegree) i, j에 해당하는 항의 계수는 c[i, j]에 저장됩니다. 만약 c의 차원이 2보다 크다면, 나머지 인덱스들은 여러 개의 계수 집합을 나타내는 데 사용됩니다.

구현 단계

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

import numpy as np
from numpy.polynomial import laguerre as L

3차원 계수 배열을 생성합니다.

c = np.arange(24).reshape(2,2,6)

생성된 배열을 출력해 내용을 확인합니다.

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의 데카르트 곱 위에서 2차원 라게르 급수를 평가합니다.

print("\nResult...\n",L.laggrid2d([1,2],[1,2], c))

전체 예제 코드

import numpy as np
from numpy.polynomial import laguerre as L

# 3차원 계수 배열 생성
c = np.arange(24).reshape(2,2,6)

# 배열 출력
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)

# laggrid2d()로 x, y 데카르트 곱 위의 2차원 라게르 급수 평가
print("\nResult...\n",L.laggrid2d([1,2],[1,2], c))

실행 결과

Our Array...
    [[[ 0  1  2  3  4  5]
    [ 6  7  8  9 10 11]]

    [[12 13 14 15 16 17]
    [18 19 20 21 22 23]]]

Dimensions of our Array...
3

Datatype of our Array object...
int64

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

Result...
    [[[ 0. -6.]
    [-12. 0.]]

    [[ 1. -6.]
    [-12. 0.]]

    [[ 2. -6.]
    [-12. 0.]]

    [[ 3. -6.]
    [-12. 0.]]

    [[ 4. -6.]
    [-12. 0.]]

    [[ 5. -6.]
    [-12. 0.]]]

결과 해석

계수 배열의 shape가 (2, 2, 6)이므로, 결과 shape는 c.shape[2:] + x.shape + y.shape, 즉 (6,) + (2,) + (2,) = (6, 2, 2)가 됩니다. 이는 마지막 축에 있는 6개의 계수 집합 각각이 [1, 2] × [1, 2]로 구성된 2×2 격자 위에서 독립적으로 평가되었음을 의미합니다. 이처럼 laggrid2d()를 활용하면 여러 계수 집합을 한 번의 호출로 효율적으로 처리할 수 있습니다.