如何根据条件绘制不同颜色的线？

Question

我有包含datatime和river_km列的数据框，我需要创建一个绘图，在其中可以查看根据datatime的river_km的变化。然后用不同的颜色标记此路径。

我的数据示例如下：

datatime    river_km    color
2018-02-05 08:22:26.000 569.5170649755595   9
2018-02-05 08:22:27.000 569.5188448645853   9
2018-02-05 08:22:28.000 569.5206201294817   9
2018-02-05 08:22:29.000 569.522379024173    9
2018-02-05 08:22:30.000 569.5241309799721   3
2018-02-05 08:22:31.000 569.5272660471763   3
2018-02-05 08:22:32.000 569.5273096158247   3

我用不同的颜色设置了新的colom：

for row in df['ref_station_id']:

    if row == 2184.0:

        color.append("blue")
    else:
        color.append("red")

df['color'] = color 

并尝试将其绘制：

df.plot( x= 'datetime', y= 'river_km', color = df.color)
plt.show()

但是我找不到一种简单的方法来给线着色。 我将不胜感激