matplotlibのlinecollectionで線を色分けしたい

Question

matplotlibにおいて、例えばx = (0,1,2,3,4) y = (0,4,3,5,7)をプロットした線グラフがあるとして、
linecollectionを利用して最初の点から次の点までは赤、次の点までは緑、次の点までは青、次の点までは黄色といったように各点間の色分けを行いたいです。
いまいちlinecollectionのcolormapの使用方法が分からないためご教授いただければ幸いです。

Answer 1

LineCollection は複数の線分をまとめたオブジェクトで、コンストラクタの第一引数 segments には、線分の一覧をリスト [line1, line2, ...] で指定します。
各 line は [(x1, y1), (x2, y2), ...] とその線分を構成する点の一覧になります。

色は線分ごとに設定できるようになっているので、x = (0,1,2,3,4) y = (0,4,3,5,7) を各区間ごとに色分けしたい場合、以下の4つの線分を作成する必要があります。

lines = [[(0, 0), (1, 4)],  # 線分1
         [(1, 4), (2, 3)],  # 線分2
         [(2, 3), (3, 5)],  # 線分3
         [(3, 5), (4, 7)]  # 線分4
        ]

サンプルコード

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.collections import LineCollection

xs = [0, 1, 2, 3, 4]
ys = [0, 4, 3, 5, 7]
lines = [[(x1, y1), (x2, y2)] for x1, y1, x2, y2 in zip(xs, ys, xs[1:], ys[1:])]
print(len(lines))
colors = ["r", "g", "b", "y"]
lc = LineCollection(lines, colors=colors)

fig, ax = plt.subplots()
ax.add_collection(lc)
ax.autoscale()
plt.show()