根据渐变符号为数据指定颜色（Python）

Question

我有两列数据，x和y。 y数据采用下面三角波的形状。如您所见，三角形有2个正梯度部分，1个长部分具有负梯度。

我想写一个程序：

• 查询垂直数组中的当前条目相对于数组中的连续条目是否具有正梯度或负梯度。
• 然后，将y数据绘制成x，其中具有正梯度的y值（及其相应的x值）使用一种颜色绘制，而负值绘制为另一种颜色。

如何在Python中做得最好？

Answer 1

filen = 'filename.txt'
x = loadtxt(fn,unpack=True,usecols=[0]) 
y = loadtxt(fn,unpack=True,usecols=[1])

n = ma.masked_where(gradient(y) < 0, y)
p = ma.masked_where(gradient(y) > 0, y)

pylab.plot(x,n,'r',x,p,'g')

我的诀窍！