グラフの色を条件で分けたい

Question

import numpy as np
import matplotlib.pyplot as plt
X = np.arange(0,11,1)
Y = np.arange(0,11,1)
F = np.array([0,0,0,0,0,0,1,1,0,1,0])

このように配列があったとして、X-Yのグラフを作成したいとします。
この時、Fが0の時は赤色、1の時は青色のようにFでグラフの色分けはできますでしょうか。
scatterではforとifでX[i], Y[i], F[i]の組みを回して色分けできたのですが、plotでもできないかなと思いました。また、なるべくforを使わないで描画したいです。

自分の書いたforとifのscatterのコードを以下に示します。

for i in range(len(X)):
    plt.scatter(X[i], Y[i], color='red') if F[i] == 0 else plt.scatter(X[i], Y[i], color='blue')
plt.show()

Answer 1

for文を使っていますが、以下のようにすればできます。

import matplotlib.pyplot as plt
import numpy as np

X = np.arange(0,11,1)
Y = np.arange(0,11,1)
F = np.array([0,0,0,0,0,0,1,1,0,1,0])

# １線分ずつ色を変えてplotする
for i in range(len(X)):
    plt.plot(X[i:i+2], Y[i:i+2], color= 'red' if F[i] == 0 else 'blue')

plt.show()

Comments

[mega_ne]

データがもし直線ではなく曲線の場合、この方法だと2点間を直線に結びそうなのでカクカクになりますか？

[8524ba23]

plot関数ではsin関数などの曲線も実際には点配列で渡すので直線で描画されます。
区間を小さくすればカクツキも目立たなくなります。

[mega_ne]

確かにそうでした、完全に忘れていました。
colorの後にifを書くことができるとは思いもよりませんでした。
できましたいつもありがとうございます。