Цвет линии Matplotlib на основе внешней переменной [duplicate]
Theres отличный учебник по этому вопросу в HelloAndroid по этому самому вопросу. Infact это был первый хит, который я получил в Google. Вы должны попробовать googling, прежде чем спрашивать здесь, поскольку это хорошая практика.
3 ответа
Я не думаю, что вы можете использовать массив цветов в plot (в документации говорится, что цвет может быть любым цветом matlab, в то время как scatter docs говорят, что вы можете использовать массив).
Однако вы можете подделать его, построив каждую строку отдельно:
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()
-
1лол. Это будет моей следующей попыткой. Спасибо за ответ! – bandito40 12 February 2016 в 23:36
-
2На самом деле это можно сделать с помощью LineCollection, но я не понимаю, как его адаптировать к моим потребностям. – bandito40 13 February 2016 в 01:59
OK. Поэтому я понял, как это сделать, используя 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()
На странице matplotlib показан пример , показывающий, как использовать 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()
