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

파이썬 NumPy legval()로 점 x에서 르장드르 급수 평가하기

파이썬 NumPy에서 특정 점 x에서 르장드르(Legendre) 급수를 평가하려면 polynomial.legendre.legval() 메서드를 사용합니다. 이 메서드는 다차원 계수 배열을 지원하며, tensor 매개변수를 통해 x의 각 차원에 맞게 계수 배열의 모양을 확장할 수 있습니다.

legval() 메서드의 주요 매개변수

1. x (첫 번째 매개변수)

급수를 평가할 지점입니다. x가 리스트나 튜플로 전달되면 자동으로 ndarray로 변환되며, 그 외의 경우에는 스칼라로 취급됩니다. 어떤 경우든 x 또는 그 요소들은 자기 자신과 계수 배열 c의 요소들 간에 덧셈과 곱셈 연산을 지원해야 합니다.

2. c (두 번째 매개변수)

계수 배열입니다. 차수 n에 해당하는 계수가 c[n]에 저장되도록 정렬되어 있습니다. c가 다차원 배열인 경우 나머지 인덱스는 여러 개의 다항식을 나타냅니다. 2차원 배열의 경우 계수가 c의 열(column)에 저장되어 있다고 생각할 수 있습니다.

3. tensor (세 번째 매개변수)

True로 설정하면 계수 배열의 모양 오른쪽에 1이 추가되어 확장됩니다. 이때 확장되는 횟수는 x의 차원 수와 같으며, 스칼라는 차원 0으로 처리됩니다. 그 결과 c의 모든 계수 열이 x의 모든 요소에 대해 평가됩니다. False로 설정하면 x가 c의 열 위로 브로드캐스팅되어 평가됩니다. 이 키워드는 c가 다차원일 때 특히 유용하며, 기본값은 True입니다.

단계별 구현 방법

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

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

다차원 계수 배열을 생성합니다.

c = np.arange(4).reshape(2,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)

legval() 메서드를 사용하여 점 [1, 2]에서 르장드르 급수를 평가합니다.

print("\nResult...\n",L.legval([1,2],c, tensor = True))

전체 예제 코드

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

# 다차원 계수 배열 생성
c = np.arange(4).reshape(2,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)

# legval() 메서드로 점 x에서 르장드르 급수 평가
print("\nResult...\n",L.legval([1,2],c, tensor = True))

실행 결과

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

Dimensions of our Array...
2

Datatype of our Array object...
int64

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

Result...
    [[2. 4.]
    [4. 7.]]

위 결과에서 볼 수 있듯이, 2x2 형태의 계수 배열 c에 대해 tensor=True 옵션을 적용하면 각 열의 계수가 점 1과 2에서 각각 평가되어 2x2 형태의 결과 배열이 반환됩니다. 이처럼 legval() 메서드는 다차원 계수 배열을 유연하게 처리할 수 있어 과학적 계산 및 수치 해석 분야에서 유용하게 활용됩니다.