Python NumPy leggrid3d()로 x, y, z 데카르트 곱 위 3D 르장드르 급수 평가하기

x, y, z의 데카르트 곱(Cartesian product) 위에서 3차원 르장드르(Legendre) 급수를 평가하려면 Python NumPy의 polynomial.legendre.leggrid3d() 메서드를 사용하면 됩니다. 이 메서드는 x, y, z의 데카르트 곱에 해당하는 점들에서 3차원 르장드르 급수의 값을 반환합니다.

계수 배열 c가 3차원보다 낮은 차원을 가지고 있다면, 3차원으로 맞추기 위해 형태(shape) 끝에 암묵적으로 1이 추가됩니다. 결과 배열의 형태는 c.shape[3:] + x.shape + y.shape + z.shape가 됩니다.

주요 매개변수

1. x, y, z

3차원 급수가 평가될 점들의 집합입니다. x, y, z의 모든 조합(데카르트 곱)에 대해 값이 계산됩니다. 인자가 리스트(list)나 튜플(tuple)로 전달되면 먼저 ndarray로 변환되며, ndarray가 아닌 경우 스칼라(scalar)로 취급됩니다.

2. c

계수(coefficient) 배열입니다. 다중 차수(multidegree) i, j, k 항의 계수가 c[i, j, k]에 저장되도록 정렬되어 있습니다. 만약 c의 차원이 3보다 크다면, 나머지 인덱스들은 여러 세트의 계수를 나타냅니다.

단계별 구현 방법

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

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

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

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

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

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

배열의 차원 수를 확인합니다.

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

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

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

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

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

leggrid3d() 메서드를 호출하여 x, y, z의 데카르트 곱 위에서 3차원 르장드르 급수를 평가합니다.

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

전체 예제 코드

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

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

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

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

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

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

# leggrid3d() 메서드로 3D 르장드르 급수 평가
print("\nResult...\n", L.leggrid3d([1, 2], [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]]]


    [[[24 25]
    [26 27]
    [28 29]
    [30 31]
    [32 33]
    [34 35]]

    [[36 37]
    [38 39]
    [40 41]
    [42 43]
    [44 45]
    [46 47]]]]

Dimensions of our Array...
4

Datatype of our Array object...
int64

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

Result...
    [[[[  552.  28911.  ]
    [  900.  46566.  ]]

    [[  972.  49765.5 ]
    [  1566.  79447.5 ]]]


    [[[  576.  29977.5 ]
    [  936.  48165.75]]

    [[  1008.  51365.25 ]
    [  1620.  81847.125]]]]

정리

NumPy의 leggrid3d() 메서드는 다차원 격자(grid) 상에서 르장드르 급수를 효율적으로 평가할 수 있는 강력한 도구입니다. 계수 배열의 차원 구조와 결과 배열의 형태 규칙(c.shape[3:] + x.shape + y.shape + z.shape)만 이해하면, 과학 계산 및 수치 해석 작업에서 다항식 근사를 손쉽게 활용할 수 있습니다.