如何删除功能不连续的连接线

时间:2012-04-30 01:40:37

标签: matplotlib

我正在绘制一些具有多个不连续性的函数。每个函数都以列表形式给出。我想只在函数连续的地方连接点。

以下是情节正在进行的简化示例。

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轻松做到这一点。

2 个答案:

答案 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)

enter image description here

答案 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()