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

Python NumPy polyvalfromroots()를 활용한 근 기반 다항식 평가 방법

Python NumPy에서 polynomial.polyvalfromroots() 메서드를 사용하면 다항식의 근(roots)으로 정의된 다항식을 특정 점 x에서 평가할 수 있습니다. 이 메서드는 계수가 아닌 근을 기준으로 다항식을 다룰 때 매우 유용합니다.

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

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

다항식을 평가할 지점입니다. x가 리스트(list)나 튜플(tuple)이면 ndarray로 자동 변환되며, 그렇지 않으면 스칼라(scalar)로 취급됩니다. 어떤 경우든 x 또는 그 요소들은 자기 자신과 r의 요소들 간에 덧셈과 곱셈을 지원해야 합니다.

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

근을 담고 있는 배열입니다. r이 다차원인 경우 첫 번째 인덱스는 근의 인덱스를 나타내고, 나머지 인덱스는 여러 개의 다항식을 구분하는 데 사용됩니다. 예를 들어 2차원 배열의 경우 각 다항식의 근들이 r의 열(column)에 저장되어 있다고 생각하면 이해하기 쉽습니다.

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

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

구현 단계

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

from numpy.polynomial.polynomial import polyvalfromroots
import numpy as np

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

c = np.arange(-2, 2).reshape(2,2)

생성한 배열을 출력합니다.

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)

polyvalfromroots() 메서드를 사용하여 근으로 정의된 다항식을 점 x에서 평가합니다.

print("\nResult...\n",polyvalfromroots(1, c))

전체 예제 코드

from numpy.polynomial.polynomial import polyvalfromroots
import numpy as np

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

# 근으로 정의된 다항식을 점 x=1에서 평가
print("\nResult...\n",polyvalfromroots(1, c))

실행 결과

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

Dimensions of our Array...
2

Datatype of our Array object...
int64

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

Result...
[3. 0.]

결과 해석

위 예제에서 입력 배열 c는 [[-2, -1], [0, 1]]이며, 각 열이 하나의 다항식의 근 집합으로 해석됩니다. 첫 번째 다항식은 근 -2와 0을 가지므로 (x+2)·x = x² + 2x이고, x=1에서의 값은 3입니다. 두 번째 다항식은 근 -1과 1을 가지므로 (x+1)·(x-1) = x² - 1이고, x=1에서의 값은 0입니다. 따라서 최종 결과는 [3. 0.]이 됩니다.