功能特定扩展系列

Question

我想实现一个特定的扩展系列，但我不知道如何做到这一点。 我的目标是能够指定系列的顺序，以避免为每个订单编写大量功能：

我试过了：

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))