python绘制ATLAS光子数据空间分布代码

在Python中，使用matplotlib库及其姐妹库如numpy和pandas可以绘制ATLAS光子数据的空间分布。以下是一个简单的示例，假设你已经有了包含光子XYZ坐标的数据：

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

# 假设你有一个名为data.csv的文件，其中包含光子X, Y, Z位置
df = pd.read_csv('data.csv')

# 提取光子坐标
x = df['X'].values
y = df['Y'].values
z = df['Z'].values

# 创建一个3D图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制点云
ax.scatter(x, y, z, c=z, cmap='viridis', alpha=0.7)  # 使用颜色编码表示Z轴高度

# 添加标签和标题
ax.set_xlabel('X (单位)')
ax.set_ylabel('Y (单位)')
ax.set_zlabel('Z (单位)')
ax.set_title('ATLAS光子在三维空间分布')

# 显示图形
plt.show()