python/matplotlib - 多色线

Question

我想创建一个带有特定条件的彩色线。基本上我希望线条在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 array of colors 
categories = numpy.array(colors) 

#create a color map with the three colors 
colormap = numpy.array([matplotlib.colors.colorConverter.to_rgb('r'),matplotlib.colors.colorConverter.to_rgb('g'),matplotlib.colors.colorConverter.to_rgb('b')]) 

#plot line 
matplotlib.axes.plot(x,y,color=colormap[categories]) 

不知道怎么弄地块（）接受的颜色数组。我总是得到关于用作颜色的格式类型的错误。尝试过十进制，十进制，字符串和浮点数。使用scatter（）完美工作。

感谢

Answer 1

我不认为你可以在plot使用的颜色数组（该文件说，颜色可以是任何颜色的matlab，而scatter文档说，你可以使用数组）。

但是，你可以伪造它通过分别绘制每一行：

import numpy 
from matplotlib import pyplot as plt 

x = range(10) 
y = numpy.random.choice(10,10) 
for x1, x2, y1,y2 in zip(x, x[1:], y, y[1:]): 
    if y1 > y2: 
     plt.plot([x1, x2], [y1, y2], 'r') 
    elif y1 < y2: 
     plt.plot([x1, x2], [y1, y2], 'g') 
    else: 
     plt.plot([x1, x2], [y1, y2], 'b') 

plt.show() 

Answer 2

确定。所以我想出了如何使用LineCollecion在轴上绘制线条。

import numpy as np 
import pylab as pl 
from matplotlib import collections as mc 

segments = [] 
colors = np.zeros(shape=(10,4)) 
x = range(10) 
y = np.random.choice(10,10) 
i = 0 

for x1, x2, y1,y2 in zip(x, x[1:], y, y[1:]): 
    if y1 > y2: 
     colors[i] = tuple([1,0,0,1]) 
    elif y1 < y2: 
     colors[i] = tuple([0,1,0,1]) 
    else: 
     colors[i] = tuple([0,0,1,1]) 
    segments.append([(x1, y1), (x2, y2)]) 
    i += 1  

lc = mc.LineCollection(segments, colors=colors, linewidths=2) 
fig, ax = pl.subplots() 
ax.add_collection(lc) 
ax.autoscale() 
ax.margins(0.1) 
pl.show() 

Answer 3

有一个example on the matplotlib page展示如何使用LineCollection来绘制五彩线。

剩下的问题是获取线条集合的颜色。所以，如果y是值进行比较，

cm = dict(zip(range(-1,2,1),list("gbr"))) 
colors = list(map(cm.get , np.sign(np.diff(y)) )) 

完整代码：

import numpy as np; np.random.seed(5) 
import matplotlib.pyplot as plt 
from matplotlib.collections import LineCollection 

x = np.arange(10) 
y = np.random.choice(10,10) 

points = np.array([x, y]).T.reshape(-1, 1, 2) 
segments = np.concatenate([points[:-1], points[1:]], axis=1) 

cm = dict(zip(range(-1,2,1),list("rbg"))) 
colors = list(map(cm.get , np.sign(np.diff(y)) )) 

lc = LineCollection(segments, colors=colors, linewidths=2) 
fig, ax = plt.subplots() 
ax.add_collection(lc) 

ax.autoscale() 
ax.margins(0.1) 
plt.show()