有没有一种简单的方法来绘制一个函数,这个函数在正反两个方向上趋于无穷大,而没有正反两端的连接?
例如,使用此代码绘制y=1/x将得到结果:import numpy as np
import matplotlib.pyplot as plt
def f(x):
return 1/x
fx_name = r'$f(x)=\frac{1}{x}$'
x=np.setdiff1d(np.linspace(-10,10,100),[0]) #to remove the zero
y=f(x)
plt.plot(x, y, label=fx_name)
plt.legend(loc='upper left')
plt.show()
但是我想要这个输出,我通过绘制两个独立的域来实现:import numpy as np
import matplotlib.pyplot as plt
def f(x):
return 1/x
fx_name = r'$f(x)=\frac{1}{x}$'
xfn=np.setdiff1d(np.linspace(-10,0,100),[0])
xfp=np.setdiff1d(np.linspace(0,10,100),[0])
yfn=f(xfn)
yfp=f(xfp)
yf = plt.plot(xfn, yfn, label=fx_name)
plt.plot(xfp, yfp, color=yf[0].get_color())
plt.legend(loc='upper left')
plt.show()
有捷径吗?
非常感谢。
解决方案
在域数组中包含零,并抑制除以零。这将强制返回的同域数组中的一个元素作为“inf”,并且不绘制“inf”。import numpy as np
import matplotlib.pyplot as plt
def f(x):
with np.errstate(divide='ignore', invalid='ignore'):
return 1/x
fx_name = r'$f(x)=\frac{1}{x}$'
x=np.linspace(-10,10,101)
y=f(x)
plt.plot(x, y, label=fx_name)
plt.legend(loc='upper left')
plt.show()
我更喜欢这种方法,因为它避免了对数组的手动操作,并且可以很容易地重用到共享同一域的其他函数(例如y=1/(x+2))。谢谢大家的贡献。