本文整理匯總了Python中PyDSTool.isfinite方法的典型用法代碼示例。如果您正苦於以下問題:Python PyDSTool.isfinite方法的具體用法?Python PyDSTool.isfinite怎麽用?Python PyDSTool.isfinite使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類PyDSTool
的用法示例。
在下文中一共展示了PyDSTool.isfinite方法的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: plot_PP_vf_custom
# 需要導入模塊: import PyDSTool [as 別名]
# 或者: from PyDSTool import isfinite [as 別名]
def plot_PP_vf_custom(gen, xname, yname, N=20, subdomain=None, scale_exp=0):
"""Draw 2D vector field in (xname, yname) coordinates of given Generator,
sampling on a uniform grid of n by n points.
Optional subdomain dictionary specifies axes limits in each variable,
otherwise Generator's xdomain attribute will be used.
For systems of dimension > 2, the non-phase plane variables will be held
constant at their initial condition values set in the Generator.
Optional scale_exp is an exponent (domain is all reals) which rescales
size of arrows in case of disparate scales in the vector field. Larger
values of scale magnify the arrow sizes. For stiff vector fields, values
from -3 to 3 may be necessary to resolve arrows in certain regions.
Requires matplotlib 0.99 or later
"""
assert N > 1
xdom = gen.xdomain[xname]
ydom = gen.xdomain[yname]
if subdomain is not None:
try:
xdom = subdomain[xname]
except KeyError:
pass
try:
ydom = subdomain[yname]
except KeyError:
pass
assert all(dst.isfinite(xdom)), "Must specify a finite domain for x direction"
assert all(dst.isfinite(ydom)), "Must specify a finite domain for y direction"
w = xdom[1]-xdom[0]
h = ydom[1]-ydom[0]
xdict = gen.initialconditions.copy()
xix = gen.funcspec.vars.index(xname)
yix = gen.funcspec.vars.index(yname)
xs = np.linspace(xdom[0], xdom[1], N)
ys = np.linspace(ydom[0], ydom[1], N)
X, Y = np.meshgrid(xs, ys)
dxs, dys = np.meshgrid(xs, ys)
dz_big = 0
vec_dict = {}
for xi, x in enumerate(xs):
for yi, y in enumerate(ys):
xdict.update({xname: x, yname: y})
dx, dy = gen.Rhs(0, xdict)[[xix, yix]]
# note order of indices
dxs[yi,xi] = dx
dys[yi,xi] = dy
dz = np.linalg.norm((dx,dy))
if dz > dz_big:
dz_big = dz
plt.quiver(X, Y, dxs, dys, angles='xy', pivot='middle', units='inches',
scale=dz_big*max(h,w)/(10*np.exp(2*scale_exp)), lw=0.01/np.exp(scale_exp-1),
headwidth=max(2,1.5/(np.exp(scale_exp-1))),
#headlength=2*max(2,1.5/(exp(scale_exp-1))),
width=0.001*max(h,w), minshaft=2, minlength=0.001)
ax = plt.gca()
print("xdom: ", xdom)
print("ydom: ", ydom)
ax.set_xlim(xdom)
ax.set_ylim(ydom)
plt.draw()