第三类完全椭圆积分的Scipy、Python与Mathematica计算疑问

近期计算第三类完全椭圆积分（形式为PI(m,n)）时，发现Python的scipy、mpmath库与Mathematica的实现存在明显差异：mpmath.ellippi返回复数，scipy.special.elliprj返回4元组，Mathematica的EllipticPi仅返回单个数值。相关测试代码如下：

Python测试代码 #

import scipy as scp
from mpmath import ellippi, ellipk
import numpy as np

r = 1
rp = 2
z = 5
zp = 7

gamma = zp - z
a = np.sqrt(gamma**2 + (rp + r)**2)
c = np.sqrt(gamma**2 + r**2)
n1 = np.sqrt(2*r/np.abs(r - c))
k = np.sqrt(4*r*rp/a**2)
sn = ellippi(n1**2, k)

def ellippi(n, m):
  return scp.special.elliprf(0., 1. - m, 1.) + (n / 3.) * scp.special.elliprj(0., 1. - m, 1., 1. - n)

print(sn, ellippi(n1**2, k))

Mathematica测试代码 #

r = 1;
rp = 2;
z = 5;
zp = 7;
gamma = zp - z;
a = Sqrt[gamma^2 + (rp + r)^2];
b = rp + r;
c = Sqrt[gamma^2 + r^2];
n1 = Sqrt[2*r/Abs[(r - c)]];
k = Sqrt[4*r*rp/a^2];
N[EllipticPi[n1^2, k]]

疑问解答 #

1. mpmath.ellippi与scipy.special.elliprj的区别是什么？

• mpmath.ellippi是直接实现第三类完全椭圆积分的专用函数，参数定义和Mathematica的EllipticPi对齐（对应PI(n, k)，k为模数），会直接返回积分结果；当输入参数导致积分存在虚部时，会返回复数结果。
• scipy.special.elliprj是底层通用辅助函数，用于计算雅可比形式的四个基本椭圆积分（RF, RD, RJ, RC），本身不直接输出第三类椭圆积分，需要用户根据第三类椭圆积分的数学定义，将其返回的多个值组合计算才能得到目标结果。

2. 是否应使用Scipy？如何将4元组转换为单个数值？

• 选择Scipy还是mpmath取决于场景：如果需要高性能计算、兼容NumPy生态，Scipy更合适；如果需要高精度符号计算或处理复数场景，mpmath是更好的选择。
• Scipy的elliprj返回的4元组是四个基本椭圆积分的结果，要得到第三类完全椭圆积分，需按照数学公式组合这些值，就像你代码中实现的自定义ellippi函数那样，通过elliprf加elliprj结果的加权计算，得到单个的第三类完全椭圆积分值。

3. 为何Scipy返回4个数值而Mathematica仅返回1个？

• Mathematica的EllipticPi是封装好的高阶函数，内部已经完成了基本椭圆积分的组合计算，直接返回第三类完全椭圆积分的最终结果，所以输出单个数值。
• Scipy的elliprj是底层通用函数，设计目的是一次性提供多个基础椭圆积分的计算结果，让用户可以灵活组合出不同类型的椭圆积分（第一、第二、第三类），而非仅针对第三类做封装，因此返回多个值，需要用户自行组合得到目标积分。

内容的提问来源于stack exchange，提问作者user8268123