三重嵌套和Numpy

Question

我正试图对一个物理问题进行积分，而我编写的代码给出的结果是原来的10倍。我想知道是否有人能给我指出正确的方向，这是我可怕的三连环还是其他什么出了问题

我正在做这个计算。（It's from this paper about calculating the ground state energy of a Lithium atom using Hylleraas co-ordinates if you're interested !!）

这是本文的相关部分，我将在下面解释我是如何将其分解的

长话短说，为了得到积分的值，无穷和（5）在q的10个值之后被截断。公式T（q）在（6）中给出，并且是三个值上的三重嵌套和：k_1_2，k_2_3&；k_3_1

这是我的T（q）代码：

def T_q(j1, j2, j3, j_1_2, j_2_3, j_3_1, alpha, beta, gamma,q):

    '''
    T_q formula for I integral summation 

    '''
    #print("q", q )
    L_1_2 = 1/2 * (j_1_2 +1) #sets adjusted values of j12 etc.  
    #print(L_1_2, "L12")
    L_2_3 = 1/2 * (j_2_3 +1)
    #print(L_2_3, "l23")
    L_3_1 = 1/2 * (j_3_1 +1)
    #print(L_3_1, "L31")

    j_1 = j1 +2
    j_2 = j2 +2
    j_3 = j3 +2



    t_q = 0


    for k_1_2 in np.arange(L_1_2 + 1):  # Triple for loop for the triple sum 
        #print("k_1_2", k_1_2)
        for k_2_3 in np.arange(L_2_3 + 1):
        #    print("k_2_3", k_2_3)
            for k_3_1 in np.arange(L_3_1 + 1):
            #    print("new loop")
                #print("k_3_1", k_3_1)

                W_mess = (W_integral((j_1 + 2*q + 2*k_1_2 + 2*k_3_1), (j_2 + j_1_2 - 2*k_1_2 + 2*k_2_3), (j_3 + j_2_3 -2*q -2*k_2_3 + j_3_1 - 2*k_3_1),alpha, beta, gamma) +
                W_integral((j_1 + 2*q + 2*k_1_2 + 2*k_3_1), (j_3 + j_3_1 - 2*k_3_1 + 2*k_2_3), (j_2 + j_1_2 -2*q -2*k_1_2 + j_2_3 - 2*k_2_3),alpha, gamma, beta) +
                W_integral((j_2 + 2*q + 2*k_1_2 + 2*k_2_3), (j_1 + j_1_2 - 2*k_1_2 + 2*k_3_1), (j_3 + j_2_3 -2*q -2*k_2_3 + j_3_1 - 2*k_3_1),beta, alpha, gamma) +
                W_integral((j_2 + 2*q + 2*k_1_2 + 2*k_2_3), (j_3 + j_2_3 - 2*k_2_3 + 2*k_3_1), (j_1 + j_1_2 -2*q -2*k_1_2 + j_3_1 - 2*k_3_1),beta, gamma, alpha) +
                W_integral((j_3 + 2*q + 2*k_2_3 + 2*k_3_1), (j_1 + j_3_1 - 2*k_3_1 + 2*k_1_2), (j_2 + j_1_2 -2*q -2*k_1_2 + j_2_3 - 2*k_2_3),gamma, alpha, beta) +
                W_integral((j_3 + 2*q + 2*k_2_3 + 2*k_3_1), (j_2 + j_2_3 - 2*k_2_3 + 2*k_1_2), (j_1 + j_1_2 -2*q -2*k_1_2 + j_3_1 - 2*k_3_1),gamma, beta, alpha))

                t_q +=   (1/((2*q+1)**2)) * C_constant(j_1_2,q,k_1_2) * C_constant(j_2_3,q,k_2_3) * C_constant(j_3_1,q,k_3_1) * W_mess
                #print("t_q, ",t_q)
    #print("t_q final",t_q)
    return t_q

（请原谅打印函数，我使用这些函数是为了确保每次迭代触发的值都是正确的-它们是我所能看到的）

每个函数都有一个常量值，其公式由（4）给出，我使用这个python函数计算：

def C_constant(j,q,k):

    '''

    Calculates C constant

    '''


    S_q_j = np.minimum( (q-1), (j+1)/2 )  # takes minimum

    constant_term = (2*q+1)/(j+2)

    binomial_term = sc.binom(j+2,(2*k+1))

    product = mp.nprod(lambda t: ((2*k + 2*t -j )/(2*k +2*q - 2*t +1)), [0,S_q_j] )

    numpy_product = np.double(product)
    C = constant_term * binomial_term * numpy_product
    return C

它依赖于资本pi积、二项式系数和乘积，但非常简单，我无法发现任何错误

它还依赖于大量的W_积分加在一起。我相信它会为输入的任何值计算正确的值：我不太确定输入的值是否正确（因此打印语句）

这是W代码

def W_integral(l,m,n,alpha,beta,gamma):

    '''
    W integral taken from this paper https://journals.aps.org/pra/abstract/10.1103/PhysRevA.52.3681
    Asks for l m n values + alpha beta gamma  and returns equation (7), in said paper
    Checked against Matlab code
    '''

    constant = np.math.factorial(l)/((alpha +beta +gamma)**(l+m+n+3))
    W_sum  =  mp.nsum(lambda p: ((np.math.factorial(l+m+n+p+2))/((l+m+2+p)*np.math.factorial(l+1+p))  * ((alpha/(alpha +beta +gamma))**p)) *  constant * mp.hyp2f1(1,l+m+n+p+3,l+m+p+3,(alpha+beta)/(alpha +beta +gamma)) ,[0,mp.inf])
    numpy_W= np.double(W_sum)
    return numpy_W

然后对每个T（q）值求和，得出该函数的最终结果：

def I_integral(j1, j2, j3, j_1_2, j_2_3, j_3_1, alpha, beta, gamma):

    '''
    Takes values for power of electron co-ordinates and returns the value of  I "
    '''


    N = 10


    I_0_N =  ((4*np.pi)**3)  * np.array([T_q(j1, j2, j3, j_1_2, j_2_3, j_3_1, alpha, beta, gamma,q) for q in np.arange(N+1)]).sum()
    #print("I_0_N before constant")

    return I_0_N

问题是，当前，与此表相比，我的I_integral(0,0,0,-1,-1,1,1,1,1)值大约是此表给出值的10倍：

最后完成的T（q）和乘以（64*pi^3）（~2000）。当我研究输出时，第一个值似乎不正确

这是因为我的射程错了吗

我意识到这是一个相当有意义的问题，但我将非常感谢任何帮助

Answer 1

你有没有尝试过使用一些数值方法，比如欧拉的向前和向后方法

用欧拉向后法推导：

让我们考虑^ {CD1>}我们的采样时间。然后，导数的近似值为：

dX(t) / dt = [x(t) - x(t-Ts)] / Ts

如果我们将连续时间平面映射到离散时间z平面（即s=e^（s*Ts）），我们得到：

dX[k] = [x[k] - x[k-1]] / Ts，其中k是离散时间瞬间

让我们考虑下面的信号作为一个例子：

X = np.linspace(0,100,100000)
y = np.sin(X*0.1)

然后，在python中，我们可以构建如下函数：

def euler_backard_method(X, Ts):
    """
    Computes the Euler's Backwards Method Numerical Derivative.

    Arguments:
        X: an input array
        Ts: the sampling time

    Output:
        the derivative of X
    """
    return [(X[idx]-X[idx-1])/Ts for idx in range(1,len(X))]

对于Ts=0.001（高频），我们得到X（euler_backard_method(X=y, Ts=1)）的以下输出：

派生示例

我们可以以同样的方式构建集成

集成

• 转发方法：s <- (z-1) / Ts
• 反向方法：s <- (z-1) / Ts * z
• 梯形法：s <- 2 * (z-1) / Ts * (z+1)

反向方法将变成：u[k] = u[k-1] + Ts * x[k]，其中u[k]是X的集成输出。相应的职能是：

def backward_integration(X, Ts):
    """
    Numerical Integration using backward method.

    Arguments:
        X: the input data
        Ts: the sampling period

    Output:
        The derivative of X
    """
    U = []
    u = 0
    for idx in range(len(X)):
        u+=Ts*X[idx]
        U.append(u)
    return U

注意：采样时间对结果有很大影响，正如您所指出的，在采样时间中，可能会得到一个大10倍的导数