使用matplotlib根据日期更改图表线的颜色

时间:2012-03-29 20:35:41

标签: python graph matplotlib

我正在密谋对抗日期。是否可以根据日期(在轴上)更改绘制线的颜色?

1 个答案:

答案 0 :(得分:6)

您可以定义蒙版并使用它们来区分线条所需的“线段”。

以下是一个例子。

import numpy as np
import matplotlib.pyplot as plt

# data
x = np.linspace(-10, 10, 1000)
y = np.sin(x)

# 4 segments defined according to some x properties
segment1 = (x<-5)
segment2 = (x>=-5) & (x<0)
segment3 = (x>=0) & (x<5)
segment4 = (x>=5)

plt.plot(x[segment1], y[segment1], '-k', lw=2)
plt.plot(x[segment2], y[segment2], '-g', lw=2)
plt.plot(x[segment3], y[segment3], '-r', lw=2)
plt.plot(x[segment4], y[segment4], '-b', lw=2)

plt.show()

plot