기능별 확장 시리즈

Question

특정 확장 시리즈를 만들고 싶습니다. 어떻게 할 수 있는지 잘 모르겠습니다. 내 목표는 각 주문에 대한 기능의 큰 숫자 쓰기 피하기 위해 내 일련의 순서를 지정 할 수있다 : I 시도

을 그 :

import numpy as np 

order = 3 

def expan(order) : 
    n = 0 
    expan = 0 
    add = (Kappa + n/2. + (-1)**n)*np.cos(n*theta/2.) - n/2.*np.cos((n-4)*theta/2.) 
    for i in order: 
     expan += add   
     power += 1 
return expan 

Answer 1

당신 numpy로 일을 벡터화하고 다음과 같이하면됩니다 :

def expand(order, kappa, theta): 
    n = np.arange(order) 
    return np.sum((kappa + n/2 + (-1)**n)*np.cos(n*theta/2) - 
        n/2*np.cos((n-4)*theta/2))