データの大きさごとにベクトルグラフの色を変えたい

Question

読み込みたいデータ

date,cs,dir
2015/8/8 0:00,332,290.5
2015/8/8 0:10,356,297.6
2015/8/8 0:20,387,303.7
2015/8/8 0:30,406,306.2
2015/8/8 0:40,359,301.8
2015/8/8 0:50,330,303.3
2015/8/8 1:00,374,315.5
2015/8/8 1:10,334,310.5
2015/8/8 1:20,249,292.4
2015/8/8 1:30,366,304.2
.
.
.

前提・実現したいこと

上のような形式のデータを読み込み時系列ベクトルグラフを作成しました。
これをcsの大きさごとに色を変えて表示したいのですがどのようにすればいいのか分かりません…（10cm/sごとなど…）
どなたかご教示頂けないでしょうか。

該当のソースコード

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import math
math.pi

#データの読み込み
df = pd.read_csv('test.csv',parse_dates=[0])
date = df['date']
cs = df['cs']/10
dir = df['dir']
rad = dir/180*math.pi

#ベクトル成分計算
u = cs*np.sin(rad)
v = cs*np.cos(rad)

#ベクトルの起点
time_len = len(date)
x = np.arange(time_len)
y = np.zeros(len(x))

#グラフサイズ
fig, ax = plt.subplots(figsize=(15, 5))

# X軸の目盛りを設定する。
dt_labels = np.array([d.strftime('%Y/%m/%d %H:%M') for d in date])
ax.set_xlim(-20, 50)
ax.set_xticks(x[::30])
ax.set_xticklabels(dt_labels[::30], rotation=45, fontsize='small')

# y軸の目盛りを設定する。
ax.set_ylim(-55, 55)
plt.yticks(np.arange(-50, 50.1, 10))

#ベクトル表示
q = ax.quiver(x, y, u, v, width=0.002, scale_units='y', scale=1,
              headlength=0, headwidth=0, headaxislength=0)

#y=0にラインを表示,vlineで縦線
ax.axhline(color='black', linewidth=1)

# x, y軸にラベルを表記
ax.set_xlabel('datetime', fontsize=13)
ax.set_ylabel('current (cm/s)', fontsize=13)

#pngとして保存
plt.subplots_adjust(left=0.125, right=0.9, bottom=0.25, top=0.9)
plt.savefig('test.png')

試したこと

Answer 1

Axes.quiverはCというオプションを取りますから、これを渡すのが一番簡単かと思います。

matplotlib.axes.Axes.quiver — Matplotlib 3.1.1 documentation

#ベクトル表示
q = ax.quiver(x, y, u, v, cs,  # csを位置引数Cに追加
              width=0.002, scale_units='y', scale=1,
              headlength=0, headwidth=0, headaxislength=0,
              cmap="winter")  # cmapは適当に選んでください
fig.colorbar(q, ax=ax) # カラーバーはあったほうが良い