根據漸變的符號（Python）爲數據分配顏色

Question

我有兩列數據，x和y。 y數據呈現下面三角波的形狀。正如你所看到的，三角形有2個正梯度部分和1個更長部分的負梯度。

我想寫一個程序：

在垂直陣列的當前條目

• 查詢是否具有相對於陣列中的連續項的正或負梯度。
• 然後，將y數據繪製爲x，其中具有正梯度（及其各自的x值）的y值使用一種顏色繪製，負繪製使用另一種顏色繪製。

在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') 

對我來說是否有把戲！