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

Python NumPy chebval()로 브로드캐스트된 점 x에서 체비쇼프 급수 평가하기

점 x에서 체비쇼프(Chebyshev) 급수를 평가하려면 Python NumPy의 chebyshev.chebval() 메서드를 사용합니다. 이 메서드는 계수 배열과 평가 지점을 받아 해당 지점에서의 급수 값을 계산해 줍니다.

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

1. x (평가 지점)

x가 리스트(list)나 튜플(tuple)이면 ndarray로 변환되며, 그렇지 않으면 변경되지 않은 채 스칼라로 처리됩니다. 어느 경우든 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 chebyshev as C

계수로 사용할 다차원 배열을 생성합니다.

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

chebyshev.chebval() 메서드를 사용해 점 [1, 2]에서 체비쇼프 급수를 평가합니다. 여기서 tensor=False로 설정하여 x가 계수 배열의 열 위로 브로드캐스트되도록 합니다.

print("\nResult (chebval)...\n",C.chebval([1,2],c,tensor=False))

전체 예제 코드

import numpy as np
from numpy.polynomial import chebyshev as C

# 계수로 사용할 다차원 배열 생성
c = np.arange(6).reshape(3,2)

# 배열 출력
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)

# chebval() 메서드로 점 x에서 체비쇼프 급수 평가
print("\nResult (chebval)...\n",C.chebval([1,2],c,tensor=False))

실행 결과

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

Dimensions of our Array...
2

Datatype of our Array object...
int64

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

Result (chebval)...
    [ 6. 42.]

실행 결과를 보면 3×2 형태의 계수 배열 c에 대해, tensor=False 옵션 덕분에 점 x = [1, 2]가 각 열에 브로드캐스트되어 각 열별로 독립적인 체비쇼프 급수 평가 결과인 [6., 42.]가 반환된 것을 확인할 수 있습니다.