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

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

르장드르(Legendre) 다항식의 유사 Vandermonde 행렬을 생성하려면 Python NumPy의 polynomial.legvander() 메서드를 사용하면 됩니다.

이 메서드는 유사 Vandermonde 행렬을 반환합니다. 반환되는 행렬의 형태(shape)는 x.shape + (deg + 1,)이며, 마지막 인덱스는 해당 르장드르 다항식의 차수를 나타냅니다. dtype은 변환된 x와 동일하게 설정됩니다.

주요 매개변수

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

deg: 결과로 생성될 행렬의 최대 차수입니다.

Vandermonde 행렬은 다항식 회귀 분석, 곡선 피팅, 수치 적분 등 다양한 수치 계산에서 활용되는 기본적인 행렬 구조입니다. 르장드르 다항식은 직교 다항식의 한 종류로, 물리학과 공학 분야의 근사 문제에서 널리 사용됩니다.

단계별 진행 과정

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

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

배열을 생성합니다.

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)

르장드르 다항식의 유사 Vandermonde 행렬을 생성하려면 polynomial.legvander() 메서드를 사용합니다.

print("\nResult...\n",L.legvander(x, 2))

전체 예제 코드

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

# 배열 생성
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)

# polynomial.legvander() 메서드로 유사 Vandermonde 행렬 생성
print("\nResult...\n",L.legvander(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. -0.5]
   [ 1. 1. 1. ]
   [ 1. -1. 1. ]
   [ 1. 2. 5.5]]

실행 결과를 보면 입력 배열 x의 각 원소에 대해 0차부터 2차까지의 르장드르 다항식 값이 열 단위로 계산되어 4×3 크기의 행렬이 생성된 것을 확인할 수 있습니다. 첫 번째 열은 항상 1(0차 다항식), 두 번째 열은 x값 그대로(1차 다항식), 세 번째 열은 (3x²−1)/2 공식에 따른 2차 르장드르 다항식의 값입니다.