3次元の散布図が描きたい

Question

前提・実現したいこと

Test.csvのうちA列をX軸， B列をY軸，C列をZ軸，D列を凡例とした3次元の散布図を描きたいのですが，以下のエラーメッセージが出てきました．これを解決し，描画に持っていくまでの過程についてお知恵をお貸し頂けますと幸いです．宜しくお願い致します．

該当のソースコード

#!/usr/bin/python
# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
from mpl_toolkits.mplot3d import Axes3D
df = pd.read_csv("Test.csv")

fig, ax = plt.subplots()
ax.set_xlim(4, 12)  # x軸の表示範囲
ax.set_ylim(8, 13)  # y軸の表示範囲
ax.set_zlim(100, 110)  # z軸の表示範囲
ax.set_xlabel("A", fontsize=10)  # x軸ラベル
ax.set_ylabel("B", fontsize=10)  # y軸ラベル
ax.set_xlabel("C", fontsize=10)  # x軸ラベル

ax.grid()  # 目盛線の表示
ax.tick_params(labelsize=10)  # 目盛線のラベルサイズ

sns.scatterplot(x="A", y="B", z="C", hue="D", data=df, ax=ax)
#ax.get_legend().remove()
ax.legend(bbox_to_anchor=(1.0, 0.05), loc='lower left', fontsize=10, ncol=1, borderaxespad=0, frameon =False)

sns.set_style("darkgrid")

#グラフの枠を作っていく
fig = plt.figure()
ax = Axes3D(fig)

ax.plot(sns.scatterplot,marker="o",linestyle='None')

#最後に.show()を書いてグラフ表示
plt.show()

エラーメッセージ

Traceback (most recent call last):
  File "csv_3D_test.py", line 12, in <module>
    ax.set_zlim(100, 110)  # z軸の表示範囲
AttributeError: 'AxesSubplot' object has no attribute 'set_zlim'

使用ファイル1

A,B,C,D
4.68,11.85431,106,α
4.73,12.50193,109,β
4.87,12.52497,109,γ
4.87,12.48392,109,α
4.87,12.48392,109,γ
4.87,12.48392,109,α
4.87,12.48392,109,β
4.87,12.48392,109,β
4.87,12.51094,109,θ
4.87,12.52497,109,θ

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

macOS10.15.4 Python3.7.3 Atom

Comments

[meg_]

エラーメッセージに「ax.set_zlim(8, 12) # z軸の表示範囲」とありますが、これは質問のコードにはありません。実際に実行したコードとエラーメッセージを提示いただかないと回答出来ません。

[meg_]

下記部分でもエラーが出ますが、このコードは何を参考に書かれたコードですか？
> sns.scatterplot(x="A", y="B", z="C", hue="D", data=df, ax=ax)
> AttributeError: 'PathCollection' object has no property 'z'

[Male_soprano]

正直に申しますと，何も参考にする事なく従来の2Dのグラフを描画するコードに何も考えず追記しただけです．(Documentを見たら2Dのみに対応している事に気づきました)

Answer 1

下記コードでエラーは出なくなりましたが、質問者さんが望むグラフになっていますでしょうか？

import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
from mpl_toolkits.mplot3d import Axes3D
df = pd.read_csv("Test.csv")

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

ax.set_xlim(4, 12)  # x軸の表示範囲
ax.set_ylim(8, 13)  # y軸の表示範囲
ax.set_zlim(100, 110)  # z軸の表示範囲
ax.set_xlabel("A", fontsize=10)  # x軸ラベル
ax.set_ylabel("B", fontsize=10)  # y軸ラベル
ax.set_xlabel("C", fontsize=10)  # x軸ラベル

ax.grid()  # 目盛線の表示
ax.tick_params(labelsize=10)  # 目盛線のラベルサイズ

for i in df['D'].unique():
    ax.plot(df[df['D']==i]['A'], df[df['D']==i]['B'], df[df['D']==i]['C'], marker="o", linestyle='None', label=i)

ax.legend(bbox_to_anchor=(1.0, 0.05), loc='lower left', fontsize=10, ncol=1, borderaxespad=0, frameon =False)

sns.set_style("darkgrid")

#最後に.show()を書いてグラフ表示
plt.show()

Comments

[Male_soprano]

有難うございます．もしも可能であれば，D列に記載されている凡例がどこかに表示されている事と，αβγθでプロットの色が異なっているととても嬉しいです．

[meg_]

画像を差し替えました。プログラム上ではちゃんと表示されていました。

[meg_]

色を変えました。

[Male_soprano]

すみません，何度か試してみたのですが凡例が見えません．
上記のコードで出力されますでしょうか？

[meg_]

jupyter labで実行した結果が回答の画像となります。

[Male_soprano]

再現できました，有難うございます．
凡例の情報のうち，αβγθ一つずつのみに色を当てるようにするにはコードをどのように書き換えれば宜しいでしょうか？

[meg_]

> 凡例の情報のうち，αβγθ一つずつのみに色を当てる
どういうことでしょうか？ 手書きでも良いので欲しいグラフを質問に追記してください。

[Male_soprano]

失礼致しました．下記リンクの「【応用】一つのグラフに折れ線と3D散布図を重ねる」に記載されているような凡例を想定しています．
https://ct-innovation01.xyz/DL-Freetime/matplotlib05/

[Male_soprano]

確認しました．有難うございました．