matplotlib 通过第三个变量为线着色- Python

vcudknz3  于 2023-03-09  发布在  Python
关注(0)|答案(2)|浏览(130)

我有以下数据集: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
我想绘制价格与日期:

但是我想用第三个变量(在我的例子中是Date或Change)来给行着色。
有谁能帮我一下吗?
谢谢。

gpfsuwkq

gpfsuwkq1#

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

import matplotlib.cm as cm
import matplotlib.pyplot as plt

def plot_colourline(x,y,c):
    col = 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=col[i])
    im = ax.scatter(x, y, c=c, s=0, cmap=cm.jet)
    return im

这个函数将所需的属性标准化,并从jet色彩Map表中获取一种颜色。这个函数返回的PathCollection也将允许绘制一个颜色条。你可能想使用一个不同的颜色条。然后,获取当前的轴,并用不同的颜色绘制数据的不同部分。因为我正在做一个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)
im = plot_colourline(x,y,prop)
fig.colorbar(im)

dvtswwa3

dvtswwa32#

如果有帮助的话,您可以通过第三个变量为数据点着色:

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()

相关问题