Python NumPy로 르장드르 다항식의 의사 Vandermonde 행렬과 점 좌표 배열 생성하기

Python NumPy에서 르장드르(Legendre) 다항식의 의사 Vandermonde(반데르몬드) 행렬을 생성하려면 legendre.legvander2d() 메서드를 사용하면 됩니다. 이 메서드는 의사 Vandermonde 행렬을 반환하며, 반환되는 행렬의 형태(shape)는 x.shape + (deg + 1,)입니다. 마지막 인덱스는 해당 르장드르 다항식의 차수(degree)를 나타내며, dtype은 변환된 x와 동일하게 유지됩니다.

주요 매개변수 이해하기

x, y: 점(point) 좌표의 배열로, 모두 동일한 형태(shape)여야 합니다. 요소 중 하나라도 복소수가 포함되어 있으면 dtype은 complex128로, 그렇지 않으면 float64로 자동 변환됩니다. 스칼라 값은 1차원 배열로 변환됩니다.

deg: 최대 차수를 지정하는 목록으로, [x_deg, y_deg] 형태로 입력합니다.

단계별 구현 방법

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

먼저 필요한 라이브러리를 가져옵니다.

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

2단계: 점 좌표 배열 생성

numpy.array() 메서드를 사용하여 모두 동일한 형태의 점 좌표 배열을 생성합니다.

x = np.array([1, 2])
y = np.array([3, 4])

3단계: 배열 출력하기

생성된 두 배열을 화면에 출력해 확인합니다.

print("Array1...\n",x)
print("\nArray2...\n",y)

4단계: 데이터 타입 확인

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

print("\nArray1 datatype...\n",x.dtype)
print("\nArray2 datatype...\n",y.dtype)

5단계: 배열의 차원 확인

두 배열의 차원(ndim)을 각각 확인합니다.

print("\nDimensions of Array1...\n",x.ndim)
print("\nDimensions of Array2...\n",y.ndim)

6단계: 배열의 형태 확인

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

print("\nShape of Array1...\n",x.shape)
print("\nShape of Array2...\n",y.shape)

7단계: 의사 Vandermonde 행렬 생성

legendre.legvander2d() 메서드를 호출하여 르장드르 다항식의 의사 Vandermonde 행렬을 생성합니다.

x_deg, y_deg = 2, 3
print("\nResult...\n",L.legvander2d(x,y, [x_deg, y_deg]))

전체 예제 코드

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

# numpy.array() 메서드로 동일한 형태의 점 좌표 배열 생성
x = np.array([1, 2])
y = np.array([3, 4])

# 배열 출력
print("Array1...\n",x)
print("\nArray2...\n",y)

# 데이터 타입 출력
print("\nArray1 datatype...\n",x.dtype)
print("\nArray2 datatype...\n",y.dtype)

# 두 배열의 차원 확인
print("\nDimensions of Array1...\n",x.ndim)
print("\nDimensions of Array2...\n",y.ndim)

# 두 배열의 형태 확인
print("\nShape of Array1...\n",x.shape)
print("\nShape of Array2...\n",y.shape)

# legendre.legvander2d() 메서드로 르장드르 다항식의 의사 Vandermonde 행렬 생성
x_deg, y_deg = 2, 3
print("\nResult...\n",L.legvander2d(x,y, [x_deg, y_deg]))

실행 결과

Array1...
    [1 2]

Array2...
    [3 4]

Array1 datatype...
int64

Array2 datatype...
int64

Dimensions of Array1...
1

Dimensions of Array2...
1

Shape of Array1...
(2,)

Shape of Array2...
(2,)

Result...
    [[ 1. 3. 13. 63. 1. 3. 13. 63. 1. 3.
     13. 63. ]
   [ 1. 4. 23.5 154. 2. 8. 47. 308. 5.5 22.
    129.25 847. ]]

위 실행 결과에서 볼 수 있듯이, 정수형(int64)으로 생성된 배열이 함수 내부에서 float64로 자동 변환되어 계산된 것을 확인할 수 있습니다. 반환된 행렬은 각 점(x, y)에 대해 x 차수 0~2, y 차수 0~3에 해당하는 르장드르 다항식의 값들을 포함하고 있으며, 이는 2차원 데이터 피팅(fitting)이나 다항식 근사 작업에 유용하게 활용됩니다.