matplotlibで３D散布図のｚ軸の値に対してカラーバーを追加

Question

matplotlibで３D散布図を作成して，ｚ軸の値に対してカラーバーを追加，色分けしたい．
よろしくお願い致します．

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

data_set = np.loadtxt(
    fname="test.csv",
    dtype="float",
    delimiter=",",
)

x1 = data_set[:,0]
y1 = data_set[:,1]
z1 = data_set[:,2]


#axオブジェクト作成
fig = plt.figure(figsize=(8, 8))
ax = fig.add_subplot(1,1,1, projection='3d')
#軸ラベル
ax.set_xlabel("T")
ax.set_ylabel("S")
ax.set_zlabel("C")

ax.set_xlim(1000, 1700)
ax.set_ylim(10, 100)
ax.set_zlim(-0.1, 0.3)

map=ax.scatter(x1, y1, z1, "o", color="blue")


#グラフの表示
plt.show()

Answer 1

map=ax.scatter(x1, y1, z1, "o", c=z1, cmap='gray')のようにc=z1でz値での色、cmapで色の範囲を指定すればよいです。

Comments

[tadapolice]

ありがとうございました．
勉強不足でした．

Answer 2

ax.set_xlim(1000, 1700)
ax.set_ylim(10, 100)
ax.set_zlim(-0.1, 0.3)

# scatter plot and colorbar
cmap = plt.colormaps.get_cmap('coolwarm')
maps = ax.scatter(x1, y1, z1, marker='o', c=z1, cmap=cmap)
maps.set_clim(*ax.get_zlim())
fig.colorbar(maps, shrink=0.4)

#グラフの表示
plt.show()

Comments

[tadapolice]

ありがとうございます．勉強不足でした．