Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python NumPy로 점 (x, y, z)에서 3차원 체비쇼프 급수 평가하기 – 4D 계수 배열 활용법

Python NumPy에서 polynomial.chebval3d() 메서드를 사용하면 점 (x, y, z)에서 3차원 체비쇼프(Chebyshev) 급수를 손쉽게 평가할 수 있습니다. 이 메서드는 x, y, z에서 대응하는 값들의 삼중 조합으로 구성된 점들에 대해 다차원 다항식의 값을 반환합니다.

매개변수 이해하기

x, y, z: 3차원 급수가 평가될 점 (x, y, z)을 나타냅니다. 세 값은 반드시 동일한 형태(shape)를 가져야 합니다. 만약 x, y, z 중 하나라도 리스트(list)나 튜플(tuple)이면 먼저 ndarray로 변환되며, ndarray가 아닌 경우에는 스칼라(scalar)로 취급됩니다.

c: 계수(coefficient)들이 담긴 배열입니다. 다중 차수(multidegree) i, j, k에 해당하는 항의 계수는 c[i, j, k]에 저장되도록 정렬됩니다. 만약 c의 차원이 3보다 크다면, 남은 인덱스들은 여러 개의 계수 집합을 열거하는 데 사용됩니다.

단계별 구현 방법

1단계: 필요한 라이브러리 임포트

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

2단계: 4D 계수 배열 생성

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

3단계: 배열 정보 확인

생성된 배열과 그 속성들을 출력하여 확인합니다.

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)

4단계: chebval3d() 메서드로 급수 평가

점 (x, y, z)에서 3차원 체비쇼프 급수를 평가하려면 polynomial.chebval3d() 메서드를 호출합니다.

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

전체 예제 코드

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

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

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

# chebval3d() 메서드로 점 (x, y, z)에서 3차원 체비쇼프 급수 평가
print("\nResult...\n", C.chebval3d([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. 148176.]
[ 576. 152631.]]

위 결과에서 볼 수 있듯이, chebval3d() 메서드는 입력된 각 점 조합에 대해 계산된 다항식 값을 2차원 배열 형태로 반환합니다. 이처럼 NumPy의 다항식 모듈을 활용하면 복잡한 다차원 체비쇼프 급수 연산도 몇 줄의 코드로 간편하게 처리할 수 있습니다.