python - 根据渐变的符号为数据分配颜色（Python）

Question

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

我想写一个程序：

• 查询垂直数组中的当前条目相对于数组中的后续条目是否具有正梯度或负梯度。
• 然后，针对 x 绘制 y 数据，其中具有正梯度的 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')

对我有用！