Python SciPy special.jnyn_zeros用法及代碼示例

本文簡要介紹 python 語言中 scipy.special.jnyn_zeros 的用法。

用法:

scipy.special.jnyn_zeros(n, nt)#

計算貝塞爾函數 Jn(x)、Jn’(x)、Yn(x) 和 Yn’(x) 的 nt 零點。

返回 4 個長度的數組nt，對應第一個nt分別為 Jn(x)、Jn’(x)、Yn(x) 和 Yn’(x) 的零點。零按升序返回。

參數 ：：

n： int

貝塞爾函數的階數

nt： int

要計算的零數 (<=1200)

返回 ：：

Jn： ndarray

Jn 的前 nt 個零

Jnp： ndarray

Jn' 的前 nt 個零

Yn： ndarray

Yn 的前 nt 個零

Ynp： ndarray

Yn' 的前 nt 個零

參考：

[1]

張善傑和金建明。 “特殊函數的計算”，John Wiley and Sons，1996 年，第 5 章。https://people.sc.fsu.edu/~jburkardt/f77_src/special_functions/special_functions.html

例子：

計算 \(J_1\) 、 \(J_1'\) 、 \(Y_1\) 和 \(Y_1'\) 的前三個根。

>>> from scipy.special import jnyn_zeros
>>> jn_roots, jnp_roots, yn_roots, ynp_roots = jnyn_zeros(1, 3)
>>> jn_roots, yn_roots
(array([ 3.83170597,  7.01558667, 10.17346814]),
 array([2.19714133, 5.42968104, 8.59600587]))

繪製 \(J_1\) 、 \(J_1'\) 、 \(Y_1\) 、 \(Y_1'\) 及其根。

>>> import numpy as np
>>> import matplotlib.pyplot as plt
>>> from scipy.special import jnyn_zeros, jvp, jn, yvp, yn
>>> jn_roots, jnp_roots, yn_roots, ynp_roots = jnyn_zeros(1, 3)
>>> fig, ax = plt.subplots()
>>> xmax= 11
>>> x = np.linspace(0, xmax)
>>> x[0] += 1e-15
>>> ax.plot(x, jn(1, x), label=r"$J_1$", c='r')
>>> ax.plot(x, jvp(1, x, 1), label=r"$J_1'$", c='b')
>>> ax.plot(x, yn(1, x), label=r"$Y_1$", c='y')
>>> ax.plot(x, yvp(1, x, 1), label=r"$Y_1'$", c='c')
>>> zeros = np.zeros((3, ))
>>> ax.scatter(jn_roots, zeros, s=30, c='r', zorder=5,
...            label=r"$J_1$ roots")
>>> ax.scatter(jnp_roots, zeros, s=30, c='b', zorder=5,
...            label=r"$J_1'$ roots")
>>> ax.scatter(yn_roots, zeros, s=30, c='y', zorder=5,
...            label=r"$Y_1$ roots")
>>> ax.scatter(ynp_roots, zeros, s=30, c='c', zorder=5,
...            label=r"$Y_1'$ roots")
>>> ax.hlines(0, 0, xmax, color='k')
>>> ax.set_ylim(-0.6, 0.6)
>>> ax.set_xlim(0, xmax)
>>> ax.legend(ncol=2, bbox_to_anchor=(1., 0.75))
>>> plt.tight_layout()
>>> plt.show()