a RG5dbS@sdZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z ddlmZmZmZddlmZmZmZmZddlmZmZmZdd lmZmZdd lmZdd l m!Z!Gd d d eZ"GdddeZ#GdddeZ$GdddeZ%GdddeZ&GdddeZ'dS)z$ Riemann zeta and related function. )Add)FunctionSsympifypiI)ArgumentIndexError expand_mul)Dummy) bernoulli factorialharmonic)re unpolarifyAbs polar_lift)log exp_polarexp)ceilingfloor)sqrt)Polyc@s:eZdZdZddZdddZddZd d Zd d Zd S)lerchphia] Lerch transcendent (Lerch phi function). Explanation =========== For $\operatorname{Re}(a) > 0$, $|z| < 1$ and $s \in \mathbb{C}$, the Lerch transcendent is defined as .. math :: \Phi(z, s, a) = \sum_{n=0}^\infty \frac{z^n}{(n + a)^s}, where the standard branch of the argument is used for $n + a$, and by analytic continuation for other values of the parameters. A commonly used related function is the Lerch zeta function, defined by .. math:: L(q, s, a) = \Phi(e^{2\pi i q}, s, a). **Analytic Continuation and Branching Behavior** It can be shown that .. math:: \Phi(z, s, a) = z\Phi(z, s, a+1) + a^{-s}. This provides the analytic continuation to $\operatorname{Re}(a) \le 0$. Assume now $\operatorname{Re}(a) > 0$. The integral representation .. math:: \Phi_0(z, s, a) = \int_0^\infty \frac{t^{s-1} e^{-at}}{1 - ze^{-t}} \frac{\mathrm{d}t}{\Gamma(s)} provides an analytic continuation to $\mathbb{C} - [1, \infty)$. Finally, for $x \in (1, \infty)$ we find .. math:: \lim_{\epsilon \to 0^+} \Phi_0(x + i\epsilon, s, a) -\lim_{\epsilon \to 0^+} \Phi_0(x - i\epsilon, s, a) = \frac{2\pi i \log^{s-1}{x}}{x^a \Gamma(s)}, using the standard branch for both $\log{x}$ and $\log{\log{x}}$ (a branch of $\log{\log{x}}$ is needed to evaluate $\log{x}^{s-1}$). This concludes the analytic continuation. The Lerch transcendent is thus branched at $z \in \{0, 1, \infty\}$ and $a \in \mathbb{Z}_{\le 0}$. For fixed $z, a$ outside these branch points, it is an entire function of $s$. Examples ======== The Lerch transcendent is a fairly general function, for this reason it does not automatically evaluate to simpler functions. Use ``expand_func()`` to achieve this. If $z=1$, the Lerch transcendent reduces to the Hurwitz zeta function: >>> from sympy import lerchphi, expand_func >>> from sympy.abc import z, s, a >>> expand_func(lerchphi(1, s, a)) zeta(s, a) More generally, if $z$ is a root of unity, the Lerch transcendent reduces to a sum of Hurwitz zeta functions: >>> expand_func(lerchphi(-1, s, a)) zeta(s, a/2)/2**s - zeta(s, a/2 + 1/2)/2**s If $a=1$, the Lerch transcendent reduces to the polylogarithm: >>> expand_func(lerchphi(z, s, 1)) polylog(s, z)/z More generally, if $a$ is rational, the Lerch transcendent reduces to a sum of polylogarithms: >>> from sympy import S >>> expand_func(lerchphi(z, s, S(1)/2)) 2**(s - 1)*(polylog(s, sqrt(z))/sqrt(z) - polylog(s, sqrt(z)*exp_polar(I*pi))/sqrt(z)) >>> expand_func(lerchphi(z, s, S(3)/2)) -2**s/z + 2**(s - 1)*(polylog(s, sqrt(z))/sqrt(z) - polylog(s, sqrt(z)*exp_polar(I*pi))/sqrt(z))/z The derivatives with respect to $z$ and $a$ can be computed in closed form: >>> lerchphi(z, s, a).diff(z) (-a*lerchphi(z, s, a) + lerchphi(z, s - 1, a))/z >>> lerchphi(z, s, a).diff(a) -s*lerchphi(z, s + 1, a) See Also ======== polylog, zeta References ========== .. [1] Bateman, H.; Erdelyi, A. (1953), Higher Transcendental Functions, Vol. I, New York: McGraw-Hill. Section 1.11. .. [2] http://dlmf.nist.gov/25.14 .. [3] https://en.wikipedia.org/wiki/Lerch_transcendent c  s|j\dkrtSjrdkrtd}t| |dd|}tj}tD]}|||7}|| |}qf| |Sj rtj}tj }dkrt krƈd88 }tfddtD}nHdkrBt d7}tfddtD}tjjg\tdtt d||dt fddtDSttrΈjdttj sd tt fvrd krtddg\ndtkrtdd g\nHt kr6td d g\n*jddtt}t|j|jg\tfd dtDStS) Nrtcs&g|]}| |qSr.0kanszrb/var/www/html/django/DPS/env/lib/python3.9/site-packages/sympy/functions/special/zeta_functions.py z.lerchphi._eval_expand_func..cs,g|]$}d||dqSrrrr rr%r&r'cs>g|]6}t|jfit|qSr)polylog_eval_expand_funcrr)hintsmrootr#zetrr%r&scsBg|]:}tdtt|t|qS)r))rrrzetar)r!pqr#rr%r&s)argsr2 is_Integerr rrZeroreversed all_coeffsdiffsubs is_RationalOnerrranger3r4rrr isinstancerr) selfr,rstartrescaddmulargr) r!r,r-r"r3r4r.r#r$r/r%r+}s`      "   4   zlerchphi._eval_expand_funcrcCs^|j\}}}|dkr*| t||d|S|dkrVt||d||t||||StdS)Nr)r5rr)r@argindexr$r#r!rrr%fdiffs  $zlerchphi.fdiffcCs|}||r|S|SdSN)r+has)r@r$r#r!targetrBrrr%_eval_rewrite_helpers zlerchphi._eval_rewrite_helpercKs||||tSrJ)rMr2r@r$r#r!kwargsrrr%_eval_rewrite_as_zetaszlerchphi._eval_rewrite_as_zetacKs||||tSrJ)rMr*rNrrr%_eval_rewrite_as_polylogsz!lerchphi._eval_rewrite_as_polylogN)r) __name__ __module__ __qualname____doc__r+rIrMrPrQrrrr%rs i< rcsPeZdZdZeddZdddZddZd d Zd d Z dfdd Z Z S)r*a Polylogarithm function. Explanation =========== For $|z| < 1$ and $s \in \mathbb{C}$, the polylogarithm is defined by .. math:: \operatorname{Li}_s(z) = \sum_{n=1}^\infty \frac{z^n}{n^s}, where the standard branch of the argument is used for $n$. It admits an analytic continuation which is branched at $z=1$ (notably not on the sheet of initial definition), $z=0$ and $z=\infty$. The name polylogarithm comes from the fact that for $s=1$, the polylogarithm is related to the ordinary logarithm (see examples), and that .. math:: \operatorname{Li}_{s+1}(z) = \int_0^z \frac{\operatorname{Li}_s(t)}{t} \mathrm{d}t. The polylogarithm is a special case of the Lerch transcendent: .. math:: \operatorname{Li}_{s}(z) = z \Phi(z, s, 1). Examples ======== For $z \in \{0, 1, -1\}$, the polylogarithm is automatically expressed using other functions: >>> from sympy import polylog >>> from sympy.abc import s >>> polylog(s, 0) 0 >>> polylog(s, 1) zeta(s) >>> polylog(s, -1) -dirichlet_eta(s) If $s$ is a negative integer, $0$ or $1$, the polylogarithm can be expressed using elementary functions. This can be done using ``expand_func()``: >>> from sympy import expand_func >>> from sympy.abc import z >>> expand_func(polylog(1, z)) -log(1 - z) >>> expand_func(polylog(0, z)) z/(1 - z) The derivative with respect to $z$ can be computed in closed form: >>> polylog(s, z).diff(z) polylog(s - 1, z)/z The polylogarithm can be expressed in terms of the lerch transcendent: >>> from sympy import lerchphi >>> polylog(s, z).rewrite(lerchphi) z*lerchphi(z, s, 1) See Also ======== zeta, lerchphi cCs0|tjurt|S|tjur&t| S|tjur6tjS|dkr||tjkrftddtdddS|dkrtddt ttdS|t dd dkrtd dtt dddddS|t dd dkrtd dtt ddddS|dt ddkrBtddtt ddddS|t dddkr|tddtt ddddS|j rtjS| tj}|rt|S|d ur|tjur|d|S|tjur|d|dS|j r|d|S| ttr,|st|tjkd kr,||t|SdS) Nr) r1r rGFT)rr=r2 NegativeOne dirichlet_etar7Halfrrrris_zeroequalsrKrrrr)clsr#r$zonerrr%evalsB      *&$$      (z polylog.evalrcCs,|j\}}|dkr$t|d||StdS)Nr)r)r5r*r)r@rHr#r$rrr%rIIs z polylog.fdiffcKs|t||dSNrr)r@r#r$rOrrr%_eval_rewrite_as_lerchphiOsz!polylog._eval_rewrite_as_lerchphicKsz|j\}}|dkr td| S|jrp|dkrptd}|d|}t| D]}|||}qLt|||St||S)Nrru) r5rr6r r>r:r r;r*)r@r,r#r$rerA_rrr%r+Rs  zpolylog._eval_expand_funccCs|jd}|jrdSdS)NrTr5r])r@r$rrr% _eval_is_zero^s zpolylog._eval_is_zerorc sddlm}|j\}}||d}|tjurJ|j|dt|jrBdndd}|j rz| |\} } Wnt t fy~|YS0| j rt|| } ||||} |||||} | tjur| S| }|g}td| D]}|| 9}||||qt|| Stt|||||S)Nr)Order-+)dirr))sympy.series.orderrir5r;rNaNlimitr is_negativer]leadterm ValueErrorNotImplementedError is_positiver _eval_nseriesremoveOr7r>appendrsuperr*)r@xr"logxcdirrinur$z0rfrnewnortermr#r __class__rr%rucs.        zpolylog._eval_nseries)r)r) rRrSrTrU classmethodrarIrdr+rhru __classcell__rrrr%r*sE .  r*c@sDeZdZdZedddZdddZddd Zd d Zdd d Z dS)r2a Hurwitz zeta function (or Riemann zeta function). Explanation =========== For $\operatorname{Re}(a) > 0$ and $\operatorname{Re}(s) > 1$, this function is defined as .. math:: \zeta(s, a) = \sum_{n=0}^\infty \frac{1}{(n + a)^s}, where the standard choice of argument for $n + a$ is used. For fixed $a$ with $\operatorname{Re}(a) > 0$ the Hurwitz zeta function admits a meromorphic continuation to all of $\mathbb{C}$, it is an unbranched function with a simple pole at $s = 1$. Analytic continuation to other $a$ is possible under some circumstances, but this is not typically done. The Hurwitz zeta function is a special case of the Lerch transcendent: .. math:: \zeta(s, a) = \Phi(1, s, a). This formula defines an analytic continuation for all possible values of $s$ and $a$ (also $\operatorname{Re}(a) < 0$), see the documentation of :class:`lerchphi` for a description of the branching behavior. If no value is passed for $a$, by this function assumes a default value of $a = 1$, yielding the Riemann zeta function. Examples ======== For $a = 1$ the Hurwitz zeta function reduces to the famous Riemann zeta function: .. math:: \zeta(s, 1) = \zeta(s) = \sum_{n=1}^\infty \frac{1}{n^s}. >>> from sympy import zeta >>> from sympy.abc import s >>> zeta(s, 1) zeta(s) >>> zeta(s) zeta(s) The Riemann zeta function can also be expressed using the Dirichlet eta function: >>> from sympy import dirichlet_eta >>> zeta(s).rewrite(dirichlet_eta) dirichlet_eta(s)/(1 - 2**(1 - s)) The Riemann zeta function at positive even integer and negative odd integer values is related to the Bernoulli numbers: >>> zeta(2) pi**2/6 >>> zeta(4) pi**4/90 >>> zeta(-1) -1/12 The specific formulae are: .. math:: \zeta(2n) = (-1)^{n+1} \frac{B_{2n} (2\pi)^{2n}}{2(2n)!} .. math:: \zeta(-n) = -\frac{B_{n+1}}{n+1} At negative even integers the Riemann zeta function is zero: >>> zeta(-4) 0 No closed-form expressions are known at positive odd integers, but numerical evaluation is possible: >>> zeta(3).n() 1.20205690315959 The derivative of $\zeta(s, a)$ with respect to $a$ can be computed: >>> from sympy.abc import a >>> zeta(s, a).diff(a) -s*zeta(s + 1, a) However the derivative with respect to $s$ has no useful closed form expression: >>> zeta(s, a).diff(s) Derivative(zeta(s, a), s) The Hurwitz zeta function can be expressed in terms of the Lerch transcendent, :class:`~.lerchphi`: >>> from sympy import lerchphi >>> zeta(s, a).rewrite(lerchphi) lerchphi(1, s, a) See Also ======== dirichlet_eta, lerchphi, polylog References ========== .. [1] http://dlmf.nist.gov/25.11 .. [2] https://en.wikipedia.org/wiki/Hurwitz_zeta_function NcCs~|dur ttt|df\}}nttt||f\}}|jrf|tjurLtjS|tjurf|durf||S|jr|tjur|tjS|tjurtjS|jrtj |S|tjurtj S|j rh|j rh|j rtj|t| d| d}nV|jr8|jr8t|t|}}tj|ddd|d|t||}ndS|j rV|tt||S|t|d|S|jrztj |SdSNrr))listmapr is_Numberrrnr=Infinityr]r\ComplexInfinity is_integerr6rprZr is_evenrtr rr abs)r_r$Za_r!r2BFrrr%ras<     $0z zeta.evalrcKs.|dkr |S|jd}t|ddd|S)Nrrr))r5r[r@r#r!rOrrr%_eval_rewrite_as_dirichlet_etas z#zeta._eval_rewrite_as_dirichlet_etacKs td||Srbrcrrrr%rd%szzeta._eval_rewrite_as_lerchphicCs"|jddj}|dur| SdS)Nrrrg)r@Z arg_is_onerrr%_eval_is_finite(szzeta._eval_is_finitecCsLt|jdkr|j\}}n|jd\}}|dkrD| t|d|StdS)Nr)r(r)lenr5r2r)r@rHr#r!rrr%rI-s  z zeta.fdiff)N)r)r)r) rRrSrTrUrrarrdrrIrrrr%r2sn '  r2c@s$eZdZdZeddZddZdS)r[a Dirichlet eta function. Explanation =========== For $\operatorname{Re}(s) > 0$, this function is defined as .. math:: \eta(s) = \sum_{n=1}^\infty \frac{(-1)^{n-1}}{n^s}. It admits a unique analytic continuation to all of $\mathbb{C}$. It is an entire, unbranched function. Examples ======== The Dirichlet eta function is closely related to the Riemann zeta function: >>> from sympy import dirichlet_eta, zeta >>> from sympy.abc import s >>> dirichlet_eta(s).rewrite(zeta) (1 - 2**(1 - s))*zeta(s) See Also ======== zeta References ========== .. [1] https://en.wikipedia.org/wiki/Dirichlet_eta_function cCs:|dkrtdSt|}|ts6ddd||SdSr)rr2rK)r_r#r$rrr%ra\s  zdirichlet_eta.evalcKsddd|t|Sr)r2)r@r#rOrrr%rPdsz#dirichlet_eta._eval_rewrite_as_zetaNrRrSrTrUrrarPrrrr%r[8s# r[c@s$eZdZdZeddZddZdS) riemann_xia Riemann Xi function. Examples ======== The Riemann Xi function is closely related to the Riemann zeta function. The zeros of Riemann Xi function are precisely the non-trivial zeros of the zeta function. >>> from sympy import riemann_xi, zeta >>> from sympy.abc import s >>> riemann_xi(s).rewrite(zeta) s*(s - 1)*gamma(s/2)*zeta(s)/(2*pi**(s/2)) References ========== .. [1] https://en.wikipedia.org/wiki/Riemann_Xi_function cCsdddlm}t|}|tjtjfvr*tjSt|ts`||d||d|dt|dSdSNr)gammarr)) 'sympy.functions.special.gamma_functionsrr2rr7r=r\r?r)r_r#rr$rrr%ras   zriemann_xi.evalcKs<ddlm}||d||dt|dt|dSr)rrr2r)r@r#rOrrrr%rPs z riemann_xi._eval_rewrite_as_zetaNrrrrr%rhs rc@seZdZdZedddZdS) stieltjesa Represents Stieltjes constants, $\gamma_{k}$ that occur in Laurent Series expansion of the Riemann zeta function. Examples ======== >>> from sympy import stieltjes >>> from sympy.abc import n, m >>> stieltjes(n) stieltjes(n) The zero'th stieltjes constant: >>> stieltjes(0) EulerGamma >>> stieltjes(0, 1) EulerGamma For generalized stieltjes constants: >>> stieltjes(n, m) stieltjes(n, m) Constants are only defined for integers >= 0: >>> stieltjes(-1) zoo References ========== .. [1] https://en.wikipedia.org/wiki/Stieltjes_constants NcCs|dur2t|}|tjur tjS|jr2|jr2tjS|jrz|tjurHtjS|dkrVtjS|jsbtjS|tjurz|dvrztjS|j rtjS|j r|dvrtjS|j dkrtjSdS)NrrbF) rrrnr6is_nonpositiverrr7 EulerGammais_extended_negativer]r)r_r"r!rrr%ras*    zstieltjes.eval)N)rRrSrTrUrrarrrr%rs$rN)(rUZsympy.core.addr sympy.corerrrrrsympy.core.functionrr sympy.core.symbolr %sympy.functions.combinatorial.numbersr r r $sympy.functions.elementary.complexesrrrr&sympy.functions.elementary.exponentialrrr#sympy.functions.elementary.integersrr(sympy.functions.elementary.miscellaneousrsympy.polys.polytoolsrrr*r2r[rrrrrr%s&    B510&