pythonでcsvファイルから3次元散布図をループ処理で作成したい

Question

前提・実現したいこと

歩行者の位置情報を記録したcsvファイルを3次元散布図として表示するために
以下のソースコードを作成しました。
行数および歩行者ID数が膨大なので，歩行者ID毎に散布図を作成するルーチンをループ処理で行いたいのですが，プログラミング初心者のため，苦戦しております。
お力添えいただければ幸いです。

該当のソースコード

# 3次元散布図の作成

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

df = pd.read_csv('group01.csv', names=('id', 'time', 'x', 'y', 'z'))
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# TODO:以下をループ処理でID毎にグラフ表示できるようにしたい
df01 = df.query('id=="1"')
df02 = df.query('id=="2"')
x01 = df01['x']
y01 = df01['y']
z01 = df01['z']
x02 = df02['x']
y02 = df02['y']
z02 = df02['z']
ax.scatter(x01, y01, z01, marker='o')
ax.scatter(x02, y02, z02, marker='x')

plt.show()

試したこと

色々調べてみると，list型を使うのが良いとのことでしたが，上手く組み込むことができませんでした...

Comments

[hayataka2049]

ちょっと要件がよくわからないのですが、歩行者ID枚に別の図として出力したいのでしょうか。それとも、質問文のコードのようにマーカを変えるなどして一枚の図にするのでしょうか（マーカの数なんてたかがしれてるので、本当に「膨大」なら難しいと思います）

[iktm]

ご回答ありがとうございます。
状況の説明が足りず申し訳ありません。
大体50〜100のID分を一枚の図にしたいと考えております。
ソースコードでは2人分のIDまでしか記載しておりませんが、ループ処理で一括して散布図を作成するフォーマットを作っておけば、色々と応用が効くと思った次第です。

[hayataka2049]

IDごとの点を区別する方法は考えなくて良いのでしょうか

[iktm]

正直なところmarkerもランダムに被りなく割り振ることができれば、それもやりたいと思っています。

Answer 1

こんな感じでしょうか。

for id_ in df["id"].unique():
    tmp = df.query("id==@id_")
    xt, yt, zt = tmp["x"], tmp["y"], tmp["z"]
    # あとはわからないので好きにやってください

Comments

[iktm]

ありがとうございます！
作成したい図が無事出力出来ました．
list型にこだわらなくてもスマートに解決できるのですね．
基礎を改めて勉強し直します．