特定の変数で色付けしたい（前回の質問の続き）

Question

前回の質問(https://teratail.com/questions/149742?whotofollow=)の続きで、この下記の可視化物を特定のx変数の値で色づけしたいです。

X4で色付けしたいのですが、どうすればいいでしょうか？

ちなみに、下記で試しましたが、そもそもデータフレームがTransposeされているので、できませんでした。
df_normalized.head(100).T.plot(figsize=[17,4],legend=False)

よろしくお願いします。

Answer 1

グループの数は既知であることを前提として。

色のリストをグループ分だけ作成しておき、df.groupby('x4') にてグループ毎にループを回すとよいのでは無いでしょうか

colors = ['red','blue','yellow','pink','green']

ax = plt.subplot()
for n, (_, d) in enumerate(df.groupby('x4')):
    d.T.plot(ax=ax, color=colors[n], legend=False)
plt.show()