python - 使用 x、y、z 图和第四个变量显示颜色图/图例

Question

我正在使用 Pandas，并且对编程非常陌生。我正在绘制能量沉积 (eDep) 作为其 x、y 和 z 位置的函数。到目前为止，成功地让它绘制，但它不会让我在散点图旁边绘制颜色图！任何帮助深表感谢

%matplotlib inline
import pandas as pd
import numpy as np

IncubatorBelow = "./Analysis.Test.csv"

df = pd.read_csv(IncubatorBelow, sep = ',', names['Name','TrackID','ParentID','xPos','yPos','zPos','eDep','DeltaE','Einit','EventID'],low_memory=False,error_bad_lines=False)
df["xPos"] = df["xPos"].str.replace("(","")
df["zPos"] = df["zPos"].str.replace(")","")

df.sort_values(by='Name', ascending=[False])
df.dropna(how='any',axis=0,subset=['Name','TrackID','ParentID','xPos','yPos','zPos','eDep','DeltaE','Einit','EventID'], inplace=True)

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
df['xPos'] = df['xPos'].astype(float)
df['yPos'] = df['yPos'].astype(float)
df['zPos'] = df['zPos'].astype(float)
#df10[df10['Name'].str.contains("e-")]

threedee = plt.figure().gca(projection='3d')
threedee.scatter(df["xPos"], df["yPos"], df["zPos"], c=df["eDep"], cmap=plt.cm.coolwarm)
threedee.set_xlabel("x(mm)")
threedee.set_ylabel("y(mm)")
threedee.set_zlabel("z(mm)")
plt.show()

这是剧情的样子！

它来自使用 GEANT4 的粒子物理模拟。实际文件非常大（我将 3.7GB 分成了 40 多 MB），这个图只代表了数据的一小部分。