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

Python NumPy로 다차원 배열의 점 x에서 르장드르 급수 평가하기

Python NumPy에서 다차원 배열 형태의 점 x에 대해 르장드르(Legendre) 급수를 평가하려면 polynomial.legendre.legval() 메서드를 사용합니다. 이 메서드는 르장드르 급수를 특정 지점에서 계산하여 그 결과값을 반환하는 강력한 도구입니다.

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

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

평가할 지점을 나타냅니다. x가 리스트(list)나 튜플(tuple)이면 ndarray로 자동 변환되며, 그렇지 않으면 변경되지 않은 채 스칼라(scalar)로 취급됩니다. 어떤 경우든 x 또는 그 요소들은 자기 자신과, 그리고 계수 배열 c의 요소들과의 덧셈 및 곱셈 연산을 지원해야 합니다.

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

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

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

True로 설정하면 계수 배열의 형태(shape)가 오른쪽에 1이 추가되어 확장되며, 추가되는 개수는 x의 각 차원마다 하나씩입니다. 이때 스칼라는 차원 0으로 간주됩니다. 그 결과 c의 모든 계수 열이 x의 모든 요소에 대해 평가됩니다. False로 설정하면 평가 시 x가 c의 열 위로 브로드캐스트(broadcast)됩니다. 이 키워드는 c가 다차원일 때 특히 유용하며, 기본값은 True입니다.

단계별 구현 방법

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

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

계수 배열을 생성합니다.

c = np.array([1, 2, 3])

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

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

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

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

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

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

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

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

여기서 x는 2차원 배열입니다.

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

다차원 배열의 점 x에서 르장드르 급수를 평가하기 위해 polynomial.legendre.legval() 메서드를 사용합니다.

print("\nResult...\n",L.legval(x,c))

전체 예제 코드

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

# 계수 배열 생성
c = np.array([1, 2, 3])

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

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

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

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

# x는 2차원 배열
x = np.array([[1,2],[3,4]])

# legval() 메서드로 르장드르 급수 평가
print("\nResult...\n",L.legval(x,c))

실행 결과

Our Array...
[1 2 3]

Dimensions of our Array...
1

Datatype of our Array object...
int64

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

Result...
[[ 6. 21.5]
[46. 79.5]]

위 결과에서 볼 수 있듯이, legval() 메서드는 2차원 배열 x의 각 요소 위치에서 르장드르 급수를 성공적으로 평가하여 동일한 형태의 결과 배열을 반환합니다. 이처럼 NumPy의 르장드르 다항식 모듈을 활용하면 복잡한 수치 계산도 간단하게 처리할 수 있습니다.