pythonにおいて生成したデータの図示の仕方が分かりません。

Question

前提・実現したいこと

pythonのライブラリの１つであるsklearnを用いてデータを作成し、それを図示したいです。
datasets.make_moonsを用いて2クラスのデータを生成し、それを1つの図にする。

発生している問題・エラーメッセージ

plt.plot(y[:,0],y[:,1],'+')

IndexError: too many indices for array

該当のソースコード

import tensorflow as tf
import matplotlib.pyplot as plt
from sklearn import datasets
#from sklearn.model_selection import train_test_split
#from sklearn.utils import shuffle

tf.set_random_seed(0)

N=300
X,y=datasets.make_moons(N,noise=0.3)

plt.plot(X[:,0],X[:,1],'.')
plt.plot(y[:,0],y[:,1],'+')
plt.show()

試したこと

Xだけなら図示できますが、X,y両方を図示しようとするとエラーが出てしまいます。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

y : array of shape [n_samples]

The integer labels (0 or 1) for class membership of each sample.

sklearn.datasets.make_moons — scikit-learn 0.19.1 documentation

なので、まあ色にでもしてみますか。あと散布図（scatter）を使った方が良いです。

import matplotlib.pyplot as plt
from sklearn import datasets

N=300
X,y=datasets.make_moons(N,noise=0.3)

plt.scatter(X[:,0],X[:,1],c=y)
plt.show()

Comments

[runatique]

ありがとうございます！散布図も後にやろうと思っていたので参考にさせていただきます！

Answer 2

やりたいことは、こんな感じでしょうか？

plt.plot(X[y==0, 0],X[y==0, 1], '.')
plt.plot(X[y==1, 0],X[y==1, 1], '+')
plt.show()

Comments

[hayataka2049]

なるほど、こういうことでしたか（理解が及ばなかった）

[runatique]

ありがとうございます！無事例通りの図が作成できました。
プログラミング初心者なので恥ずかしながらお聞きしたいのですが、この変化（[]の中身をこのように書く）は図にどのような影響を与えているのでしょうか。
ｙというのがプロットに相当するもので、Xはその全体の総称のようなものですか？だからｙが二つ必要で==0,1で表されているという感じでしょうか？

[hayataka2049]

magichanさんの代わりに答えてしまいますが、X[y==0, 0]はX[y==0][:,0]と書くのと結果的に等価です（パフォーマンスとかは違う可能性がありますが）。
yは点の属するクラスを0,1で表現するものなので、y==0のものだけ取り出しています。y==0はboolの配列になり、numpy配列のスライスでboolの配列を渡すとTrueのところだけ取り出してくれます。
Xはすべて点の座標です。yで対応するクラスのラベルが示されています。

[magichan]

X に入っているのが各点のポジションデータですので
X[0, 0] は 1つ目のデータの x座標
X[0, 1] は 1つ目のデータの y座標
X[1, 0] は 2つ目のデータの x座標
X[1, 1] は 2つ目のデータの y座標
…
となります。
次に、y に入っているのが各点のカテゴリデータ（１次元データ）なので
y[0] は 1つ目のデータのカテゴリー（0 or 1）
y[1] は 2つ目のデータのカテゴリー（0 or 1）
…
になります。
で、質問のコードやhayataka2049さんのコードにある
plot(X[:,0], X[:,1])
は１番目の引数に全てのデータのx座標、２番目の引数に全てのデータのy座標
を渡しておりますので、全てのデータがプロットされます
対して、私の回答の
plot(X[y==0,0], X[y==0,1])
は『カテゴリーデータが0であるデータ』を抽出してx座標とy座標を渡す。
と言う意味となります。

[magichan]

おっ、hayataka2049さんが既にコメントを書いてますね。
とりあえず
print(X)
print(y)
print(y==0)
print(X[y==0, 0])
などの表示してみると理解が進むかと思います。

[runatique]

ありがとうございます！とても分かりやすく助かります！
いろいろ試してみます！