Python NumPy로 체비셰프(Chebyshev) 다항식의 반데르몬드(Vandermonde) 행렬 생성하기

Python NumPy에서 chebyshev.chebvander() 함수를 사용하면 체비셰프(Chebyshev) 다항식의 반데르몬드(Vandermonde) 행렬을 손쉽게 생성할 수 있습니다. 이 메서드는 계산된 반데르몬드 행렬을 반환하며, 반환되는 행렬의 형태(shape)는 x.shape + (deg + 1,)입니다. 여기서 마지막 인덱스는 해당 차수의 체비셰프 다항식의 차수를 나타냅니다. 데이터 타입(dtype)은 변환된 x와 동일하게 유지됩니다.

주요 매개변수

x (a): 점(point)들의 배열입니다. 요소 중 하나라도 복소수가 포함되어 있으면 dtype은 complex128로 변환되고, 그렇지 않으면 float64로 변환됩니다. 만약 x가 스칼라 값이라면 자동으로 1차원 배열로 변환됩니다.

deg: 결과로 생성될 행렬의 최대 차수를 지정합니다.

구현 단계

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

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

배열을 생성합니다.

x = np.array([0, 1, -1, 2])

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

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

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

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

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

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

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

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

체비셰프 다항식의 반데르몬드 행렬을 생성하기 위해 chebyshev.chebvander() 함수를 호출합니다.

print("\nResult...\n",C.chebvander(x, 2))

전체 예제 코드

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

# 배열 생성
x = np.array([0, 1, -1, 2])

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

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

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

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

# 체비셰프 다항식의 반데르몬드 행렬 생성
print("\nResult...\n",C.chebvander(x, 2))

실행 결과

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

Dimensions of our Array...
1

Datatype of our Array object...
int64

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

Result...
    [[ 1. 0. -1.]
     [ 1. 1. 1.]
     [ 1. -1. 1.]
     [ 1. 2. 7.]]

결과 해석

위 실행 결과를 보면, 입력 배열 [0, 1, -1, 2]에 대해 차수 2까지의 체비셰프 다항식 T₀(x)=1, T₁(x)=x, T₂(x)=2x²−1 값들이 각 행에 계산되어 4×3 크기의 반데르몬드 행렬이 생성된 것을 확인할 수 있습니다. 예를 들어 x=2인 경우, T₀(2)=1, T₁(2)=2, T₂(2)=2·4−1=7이므로 마지막 행은 [1, 2, 7]이 됩니다. 이러한 반데르몬드 행렬은 체비셰프 다항식 기반의 최소제곱 피팅(fitting)이나 다항식 근사 문제를 풀 때 유용하게 활용됩니다.