python - 省略 matplotlib 图中的连接线，例如y = tan(x)

Question

我有图表y = tan(x)，我想删除垂直线(见下文)。

这是我的代码:

import numpy as np
import matplotlib.pyplot as plt

# Choose evenly spaced x intervals
x = np.arange(-2*np.pi, 2*np.pi, 0.1)

# plot y = tan(x)
plt.plot(x, np.tan(x))

# Set the range of the axes
plt.axis([-2*np.pi, 2*np.pi, -2, 2])

# Include a title
plt.title('y = tan(x)')

# Optional grid-lines
plt.grid()

# Show the graph
plt.show()

这是图表(包括不需要的垂直线):

我可以在不为 x 间隔设置适当间隙的情况下删除垂直线吗？

Answer 1

您可以使用 diff 检查连续数据点之间的差异然后确定差异为负的位置并将这些值替换为 NaN 以在绘制线中创建视觉中断

# Compute the tangent for each point
y = np.tan(x)

# Insert a NaN where the difference between successive points is negative
y[:-1][np.diff(y) < 0] = np.nan

# Plot the resulting discontinuous line
plt.plot(x, y)