numpy中是否有elliptictheta函數？

Question

我需要mathematica中的EllipticTheta函數，我不知道numpy中是否有相應的函數。尤其是第二種。

Answer 1

試試這個：http://www.google.com/search?q=python+elliptic+theta

mpmath有雅可比theta函數：

http://docs.sympy.org/0.7.6/modules/mpmath/functions/elliptic.html#jacobi-theta-functions

如果你想有一個量化的功能，你可以寫

import mpmath 
import numpy as np 

jtheta = np.vectorize(mpmath.jtheta, 'D') 

print(jtheta(1,2,1j*np.linspace(0,0.9,15))) 

注意np.vectorize不加速它 - 它只允許 它直接在陣列輸入上工作。 mpmath函數與 arb一起使用。精度浮動，所以它們有時可能比在機器精度fp中實現的等效功能慢。