python - 尝试在 matplotlib 中绘制双曲线会沿垂直渐近线产生线吗？

Question

我正在尝试使用 pyplot 和 matplotlib 绘制双曲线。这是我的代码：

from __future__ import division

import numpy
import matplotlib.pyplot as pyplot

x = numpy.arange(0, 1000, 0.01)
y = [10 / (500.53 - i) for i in x]
pyplot.plot(x, y, 'b')
pyplot.axis([0, 1000, -10, 10])

pyplot.show()

它产生以下图表：

如何修改我的图表以删除沿垂直渐近线延伸的那条线？

Answer 1

在绘图之前，添加这些行：

 threshold = 1000 # or a similarly appropriate threshold
 y = numpy.ma.masked_less(y, -1*threshold) 
 y = numpy.ma.masked_greater(y, threshold).

然后做

pyplot.plot(x, y, 'b')
pyplot.axis([0, 1000, -10, 10])
pyplot.show()

像往常一样。

另请注意，由于您使用的是 numpy 数组，因此不需要列表推导来计算y

In [12]: %timeit y = [10 / (500.53 - i) for i in x]
1 loops, best of 3: 202 ms per loop

In [13]: %timeit y = 10 / (500.53 - x)
1000 loops, best of 3: 1.23 ms per loop

希望这可以帮助。