3次元格子点をfor文でプロットしたい.

Question

前提・実現したいこと

3次元格子点をfor文でプロットしたい.

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

格子点が1つ目しか表示されない.

該当のソースコード

import pandas as pd
import numpy as np

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt

data = pd.read_csv("C:\sample\mesh1.csv",  header = None)
data.columns = ['X', 'Y', 'Z', 'label']
data.head()

y0 = data.iloc[0:, 3].values

X = np.copy(data.iloc[0:, 0].values)

Y = np.copy(data.iloc[0:, 1].values)

Z = np.copy(data.iloc[0:, 2].values)

XYZ = data.iloc[0:, [0,1,2]].values

fig = plt.figure()
ax = Axes3D(fig)

for idx in range(len(np.unique(y0))):
    ax.scatter(XYZ[idx, 0], XYZ[idx, 1],  XYZ[idx, 2],
            color='black', marker='x', linestyle='None')

csvの内容

[[-3 -3 -3 1]
[-3 -3 -2 1]
[-3 -3 -1 1]
...
[ 3 3 1 1]
[ 3 3 2 1]
[ 3 3 3 1]]
です. [-3, -3, -3]しかプロットされませんでした. 以下がcsvを作成した際のコードです.

lst = []
l = 1
for i in range(-3, 4):
    for j in range(-3, 4):
        for k in range(-3, 4):
            lst.append([i, j, k, l])
            

with open("C:\sample\mesh1.csv", 'w') as f:
    writer = csv.writer(f)
    writer.writerows(lst)

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

anaconda JupiterLab 1.2.6

Comments

[meg_]

len(np.unique(y0))が1になっていませんか？

Answer 1

csvデータの点を全て表示したいなら下記となります。

import pandas as pd
import numpy as np

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt

data = pd.read_csv("mesh1.csv",  header = None)
data.columns = ['X', 'Y', 'Z', 'label']
data.head()

y0 = data.iloc[0:, 3].values

X = np.copy(data.iloc[0:, 0].values)

Y = np.copy(data.iloc[0:, 1].values)

Z = np.copy(data.iloc[0:, 2].values)

XYZ = data.iloc[0:, [0,1,2]].values

fig = plt.figure()
ax = Axes3D(fig)

for idx in range(len(data)):  # 変更点
    ax.scatter(XYZ[idx, 0], XYZ[idx, 1],  XYZ[idx, 2],
            color='black', marker='x', linestyle='None')

plt.show()

Comments

[os-t]

できました. ありがとうございます.

Answer 2

y0 = data.iloc[0:, 3].valuesにて～.iloc[0:～としているので先頭１行目しか取得していません。XYZ = ～も同様です。
よって先頭の行しか出力できていません。