我正在尝试在某些条件下创建一条彩色线。基本上,我希望在 y 轴上指向下方时将线条着色为红色,指向上方时为绿色,而两者都不为蓝色。
我玩了一些我发现的类似例子,但我从来没有能够将它们转换为在轴上使用 plot() 。只是想知道如何做到这一点。
这是我到目前为止想出的一些代码:
#create x,y coordinates
x = numpy.random.choice(10,10)
y = numpy.random.choice(10,10)
#create an array of colors based on direction of line (0=r, 1=g, 2=b)
colors = []
#create an array that is one position away from original
#to determine direction of line
yCopy = list(y[1:])
for y1,y2 in zip(y,yCopy):
if y1 > y2:
colors.append(0)
elif y1 < y2:
colors.append(1)
else:
colors.append(2)
#add tenth spot to array as loop only does nine
colors.append(2)
#create a numpy …Run Code Online (Sandbox Code Playgroud)