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

Python NumPy로 르장드르 다항식의 의사 반데르몬드(Vandermonde) 행렬 생성하기

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

주요 매개변수 설명

x, y — 점 좌표를 담은 배열로, 두 배열은 서로 동일한 형태(shape)여야 합니다. 요소에 복소수가 포함되어 있는지 여부에 따라 dtype이 float64 또는 complex128로 변환되며, 스칼라 값이 입력되면 자동으로 1차원 배열로 변환됩니다.

deg — 최대 차수를 지정하는 매개변수로, [x_deg, y_deg] 형태의 리스트를 전달받습니다.

구현 단계

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

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

2단계: 점 좌표 배열 생성하기

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

x = np.array([0.1, 1.4])
y = np.array([1.7, 2.8])

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단계: 의사 반데르몬드 행렬 생성하기

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

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([0.1, 1.4])
y = np.array([1.7, 2.8])

# 배열 출력
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() 메서드로 르장드르 다항식의 의사 반데르몬드 행렬 생성
x_deg, y_deg = 2, 3
print("\nResult...\n",L.legvander2d(x,y, [x_deg, y_deg]))

실행 결과

Array1...
    [0.1 1.4]

Array2...
    [1.7 2.8]

Array1 datatype...
float64

Array2 datatype...
float64

Dimensions of Array1...
1

Dimensions of Array2...
1

Shape of Array1...
(2,)

Shape of Array2...
(2,)

Result...
    [[ 1.0000000e+00  1.7000000e+00  3.8350000e+00  9.7325000e+00
      1.0000000e-01  1.7000000e-01  3.8350000e-01  9.7325000e-01
     -4.8500000e-01 -8.2450000e-01 -1.8599750e+00 -4.7202625e+00]
   [ 1.0000000e+00 2.8000000e+00 1.1260000e+01 5.0680000e+01
     1.4000000e+00 3.9200000e+00 1.5764000e+01 7.0952000e+01
     2.4400000e+00 6.8320000e+00 2.7474400e+01 1.2365920e+02]]

정리

NumPy의 legvander2d() 메서드를 활용하면 2차원 르장드르 다항식 기저 함수 값으로 구성된 의사 반데르몬드 행렬을 손쉽게 계산할 수 있습니다. 이렇게 생성된 행렬은 다항식 회귀 분석이나 최소제곱법을 이용한 곡선 피팅 등 다양한 수치 계산 작업에서 유용하게 활용됩니다.