Python에서 3D 계수 배열로 점 (x, y)에서의 2차원 체비쇼프 급수 평가하기

점 (x, y)에서 2차원 체비쇼프(Chebyshev) 급수를 평가하려면 Python NumPy의 polynomial.chebval2d() 메서드를 사용합니다. 이 메서드는 x와 y의 대응하는 값들로 구성된 점들에서 2차원 체비쇼프 급수의 값을 반환합니다.

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

x, y

2차원 급수가 평가될 점 (x, y)입니다. 이때 x와 y는 반드시 동일한 형태(shape)를 가져야 합니다. 만약 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 chebyshev as C

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

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

배열을 출력하여 확인합니다.

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)

이제 polynomial.chebval2d() 메서드를 사용하여 점 (x, y)에서 2차원 체비쇼프 급수를 평가합니다.

print("\nResult...\n",C.chebval2d([1,2],[1,2], c))

전체 예제 코드

import numpy as np
from numpy.polynomial import chebyshev as C

# 3D 계수 배열 생성
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)

# chebval2d() 메서드로 점 (x, y)에서 2차원 체비쇼프 급수 평가
print("\nResult...\n",C.chebval2d([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...
[[ 36. 108.]
[ 40. 117.]
[ 44. 126.]
[ 48. 135.]
[ 52. 144.]
[ 56. 153.]]

위 결과에서 볼 수 있듯이, chebval2d() 메서드는 입력된 각 점 쌍 (x, y)에 대해 2차원 체비쇼프 급수의 평가값을 담은 배열을 반환합니다. 계수 배열이 3차원인 경우, 마지막 인덱스는 여러 개의 계수 세트를 나타내므로 각 세트별로 평가 결과가 계산됩니다.