matplotlibでの3次元の散布図

Question

3種類のデータ、４種類のラベルで、１２０個のデータがあります。
これを三次元の散布図にしたいのですが、120個のデータのうち、最後の30個しかプロットされません。
データは、ラベル0が30個、１が30個、2が30個、3が30個の順です。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt 
from mpl_toolkits.mplot3d import axes3d
pd.options.display.max_rows = 8
state = np.random.RandomState(0)
# csv からデータ読み込み
names = ['red', 'green', 'blue','season']
coloer = pd.read_csv('personal_coloer_data_rgb.csv', header=None, names=names)
df=coloer
label=df[['season']]
data=df[['red', 'green', 'blue']]

X=data
T=label


hue = 'season'
labels = set(df[hue])
dataset = []
for x in labels:
    xs = df[df[hue]==x]['red']
    ys = df[df[hue]==x]['green']
    zs = df[df[hue]==x]['blue']
    dataset.append((xs, ys, zs))

fig = plt.figure()

ax = fig.add_subplot(1, 1, 1, projection='3d')
for data, label in zip(dataset, labels):
    ax.scatter(xs, ys, zs, label=label)
ax.set_title('Matplot 3d scatter plot')  # タイトル
ax.set_xlabel('red')  # X軸ラベル
ax.set_ylabel('green')  # Y軸ラベル
ax.set_zlabel('blue')  # Z軸ラベル
ax.legend(loc=2, title='legend', shadow=True)  # 凡例

plt.show()

出力すると
このようになっています

使っているデータは、

red green blue season
0 1 0.99 0.96 0
1 0.98 0.93 0.78 0
2 0.94 0.92 0.87 0
・
・
・
30 0.95 0.91 0.85 1
31 0.7 0.6 0.47 1
32 0.71 0.57 0.32 1
・
・
・
60 0.97 0.97 0.95 2
61 0.6 0.64 0.69 2
62 0.32 0.38 0.48 2
・
・
・
90 1 0.87 0.96 3
91 0.85 0.89 1 3
92 0.92 0.99 0.91 3
・
・
・
119 0.92 0.99 0.91 3

---

labelのseasonが0~3までそれぞれ30個ずつあります

Comments

[hayataka2049]

この質問内容の場合、データがあった方が回答がつきやすいので、可能であれば
personal_coloer_data_rgb.csv
をテキスト形式でまるごと掲載してください。

Answer 1

for data, label in zip(dataset, labels):
    xs, ys, zs = data  # この行を追加してください
    ax.scatter(xs, ys, zs, label=label)