颜色线由第三个变量 - Python

时间:2016-04-08 17:17:49

标签: python matplotlib plot

我有以下数据集: In[55]: usdbrl Out[56]: Date Price Open High Low Change STD 0 2016-03-18 3.6128 3.6241 3.6731 3.6051 -0.31 0.069592 1 2016-03-17 3.6241 3.7410 3.7449 3.6020 -3.16 0.069041 2 2016-03-16 3.7422 3.7643 3.8533 3.7302 -0.62 0.068772 3 2016-03-15 3.7656 3.6610 3.7814 3.6528 2.83 0.071474 4 2016-03-14 3.6618 3.5813 3.6631 3.5755 2.23 0.070348 5 2016-03-11 3.5820 3.6204 3.6692 3.5716 -1.09 0.076458 6 2016-03-10 3.6215 3.6835 3.7102 3.6071 -1.72 0.062977 7 2016-03-09 3.6849 3.7543 3.7572 3.6790 -1.88 0.041329 8 2016-03-08 3.7556 3.7826 3.8037 3.7315 -0.72 0.013700 9 2016-03-07 3.7830 3.7573 3.7981 3.7338 0.63 0.000000

我想根据日期绘制价格: enter image description here

但我想用第三个变量(在我的情况下为Date或Change)为该行着色。

有人可以帮忙吗?

感谢。

2 个答案:

答案 0 :(得分:5)

我写了一个简单的函数来将给定的属性映射为颜色:

import matplotlib.cm
import matplotlib.pyplot as plt

def plot_colourline(x,y,c):
    c = cm.jet((c-np.min(c))/(np.max(c)-np.min(c)))
    ax = plt.gca()
    for i in np.arange(len(x)-1):
        ax.plot([x[i],x[i+1]], [y[i],y[i+1]], c=c[i])
    return

此函数规范化所需属性并从jet colormap获取颜色。您可能想要使用另一个。然后,获取当前轴并使用不同的颜色绘制数据的不同段。因为我正在进行for循环,所以应该避免将它用于非常大的数据集,但是,出于正常目的,它很有用。

将以下示例视为测试:

import numpy as np
import matplotlib.pyplot as plt

n = 100
x = 1.*np.arange(n)
y = np.random.rand(n)
prop = x**2

fig = plt.figure(1, figsize=(5,5))
ax  = fig.add_subplot(111)
plot_colourline(x,y,prop)

enter image description here

答案 1 :(得分:4)

您可以通过第三个变量为数据点着色,如果这样会有所帮助:

dates = [dt.date() for dt in pd.to_datetime(df.Date)]
plt.scatter(dates, df.Price, c=df.Change, s=100, lw=0)
plt.plot(dates, df.Price)
plt.colorbar()
plt.show()

enter image description here