我正在绘制一些具有多个不连续性的函数。每个函数都以列表形式给出。我想只在函数连续的地方连接点。
以下是情节正在进行的简化示例。
x=linspace(0,1,100)
y=zeros(100)
y[x<0.5] = x[x<0.5]
y[x>=0.5] = 1 + x[x>=0.5]
plot(x, y, '-o')
x = 0.5处存在不连续性,但是无论如何,绘图都将所有点与线连接起来。
我的功能当然不同。它们通常在不同的地方有几个不连续点。不连续性的标准很简单。比方说,如果函数跳跃超过0.5,我认为它在那时是不连续的。
是否有一个选项可以告诉它删除功能不连续的点之间的连接线?我记得能用gnuplot轻松做到这一点。
答案 0 :(得分:20)
使用nan将线分成多个部分:
import numpy as np
from pylab import *
x=linspace(0,1,100)
y=zeros(100)
y[x<0.5] = x[x<0.5]
y[x>=0.5] = 1 + x[x>=0.5]
pos = np.where(np.abs(np.diff(y)) >= 0.5)[0]
x[pos] = np.nan
y[pos] = np.nan
plot(x, y, '-o')
编辑:
在不连续处插入nan:
pos = np.where(np.abs(np.diff(y)) >= 0.5)[0]+1
x = np.insert(x, pos, np.nan)
y = np.insert(y, pos, np.nan)
答案 1 :(得分:0)
这是我建议绘制tan(x):
import matplotlib.pyplot as plt
from math import *
x_lim = 3*pi/2
y_lim = 5
n = 1000
X = []
Y = []
Z = []
for i in range(0,2*n):
x = -x_lim + i*x_lim/n
y = tan(x)
if y<y_lim and y>-y_lim:
X.append(x)
Y.append(y)
else:
if len(X)>0 and len(Y)>0:
Z.append([X,Y])
del X,Y
X = []
Y = []
for i in range(0, len(Z)):
plt.plot(Z[i][0],Z[i][1])
plt.grid(True)
plt.show()