【Python】 matplotlibについて

Question

x,y,r,g,bの5つの1次元配列があって
それぞれ長さは同じ
これを使って散布図を描きたいときに
例えば

ax.scatter(x, y)

とする方法がありますが
このとき点の色を上の配列r,g,bを使って
rgb値で指定したい
イメージとしては

ax.scatter(x, y, c=(r,g,b))

こんな感じで行いたいのですが
何か良い方法はないでしょうか
もちろん

for i in range(arr_len):
ax.scatter(x[i], y[i], c=(r[i],g[i],b[i]))

このようにすればひとまず目的は達成できるのですが
一気に描画する方法があれば教えていただきたいです

Answer 1

RGBは#000000形式文字列の指定が可能みたいですので、

c = ['#%02x%02x%02x' % x for x in zip(r, g, b)]
plt.scatter(x, y, c=c)

のようにr, g, bの各リストを1つのリストに変換すればできると思います。

0～1のfloatでも指定できますが、上記の方が直観的なのでそちらを紹介しました。

...

※最初の↓はあまり回答になっていなかったと思いましたので、書き直しました。

---

r, g, bをひとつの配列にまとめてしまえばできるみたいです。

下記リンクのページでは、numpy.ndarray型を使って、配列をまとめて渡してscatterする方法が紹介されています。

shapes_and_collections example code: scatter_demo.py — Matplotlib 1.5.1 documentation
http://matplotlib.org/examples/shapes_and_collections/scatter_demo.html

Comments

[A5size]

回答ありがとうございます
この方法でうまくいきました
また、URLもありがとうございます
非常に助かりました