使用PyROOT读取TTree分支遇属性错误的解决方案咨询

看到你遇到的问题了——能访问clust_pix这种简单数组分支，但访问traj这种结构化分支时触发AttributeError，这是PyROOT处理复杂分支的常见坑。下面给你几个可行的解决方案：

方案1：用MakeClass生成自动映射的类 #

这是ROOT官方推荐的方法，能自动生成对应分支结构的Python类，完美适配你的traj分支：

1. 先在PyROOT中生成类文件：

import ROOT
f = ROOT.TFile("../Ntuple/v0406_default_1012p2_101X_dataRun2_Express_v7_Fill6573_Ntuple.root")
t = f.Get("trajTree")
t.MakeClass("TrajTreeClass")

这会生成TrajTreeClass.h和TrajTreeClass.C两个文件，自动映射trajTree的所有分支结构。

2. 加载生成的类并使用：

import ROOT
# 编译并加载生成的类
ROOT.gROOT.ProcessLine(".L TrajTreeClass.C+")
from ROOT import TrajTreeClass

f = ROOT.TFile("../Ntuple/v0406_default_1012p2_101X_dataRun2_Express_v7_Fill6573_Ntuple.root")
t = f.Get("trajTree")

# 创建类实例并关联到树
my_tree = TrajTreeClass()
my_tree.Init(t)

# 遍历条目访问traj的成员
for i in range(min(10, t.GetEntries())):  # 先测试前10条
    my_tree.GetEntry(i)
    # 直接访问my_tree.beta、my_tree.validhit等成员
    print(f"Entry {i}: beta = {my_tree.beta}, validhit = {my_tree.validhit}")

方案2：手动用SetBranchAddress绑定变量 #

如果不想生成额外文件，可以手动定义对应分支结构的变量，然后绑定到分支：

import ROOT

# 定义对应traj分支结构的类
class TrajData:
    def __init__(self):
        self.validhit = ROOT.Int_t()
        self.missing = ROOT.Int_t()
        self.lx = ROOT.Float_t()
        self.ly = ROOT.Float_t()
        self.lz = ROOT.Float_t()
        self.glx = ROOT.Float_t()
        self.gly = ROOT.Float_t()
        self.glz = ROOT.Float_t()
        self.clust_near = ROOT.Int_t()
        self.hit_near = ROOT.Int_t()
        self.pass_effcuts = ROOT.Int_t()
        self.alpha = ROOT.Float_t()
        self.beta = ROOT.Float_t()
        self.norm_charge = ROOT.Float_t()
        self.d_tr = ROOT.Float_t()
        self.dx_tr = ROOT.Float_t()
        self.dy_tr = ROOT.Float_t()
        self.d_cl = ROOT.Float_t()
        self.dx_cl = ROOT.Float_t()
        self.dy_cl = ROOT.Float_t()
        self.dx_hit = ROOT.Float_t()
        self.dy_hit = ROOT.Float_t()
        self.onedge = ROOT.Int_t()
        self.lx_err = ROOT.Float_t()
        self.ly_err = ROOT.Float_t()

f = ROOT.TFile("../Ntuple/v0406_default_1012p2_101X_dataRun2_Express_v7_Fill6573_Ntuple.root")
t = f.Get("trajTree")

traj_data = TrajData()
# 把每个成员绑定到分支的对应字段
t.SetBranchAddress("traj.validhit", traj_data.validhit)
t.SetBranchAddress("traj.missing", traj_data.missing)
t.SetBranchAddress("traj.lx", traj_data.lx)
t.SetBranchAddress("traj.beta", traj_data.beta)
# 其他成员按`t.Print()`里的字段逐一绑定

# 遍历访问数据
for i in range(min(10, t.GetEntries())):
    t.GetEntry(i)
    print(f"Entry {i}: beta = {traj_data.beta}, lx = {traj_data.lx}")

方案3：直接通过分支对象访问（适合快速测试） #

如果只是临时查看数据，可以直接获取traj分支，通过GetEntry后访问其成员：

import ROOT

f = ROOT.TFile("../Ntuple/v0406_default_1012p2_101X_dataRun2_Express_v7_Fill6573_Ntuple.root")
t = f.Get("trajTree")
traj_branch = t.GetBranch("traj")

# 先加载一个条目初始化分支结构
traj_branch.GetEntry(0)
# 直接访问分支的成员
print(f"First entry beta: {traj_branch.beta}")

# 遍历所有条目
for i in range(min(10, t.GetEntries())):
    traj_branch.GetEntry(i)
    print(f"Entry {i}: validhit = {traj_branch.validhit}")

错误原因说明 #

你能访问clust_pix是因为它是简单的数值数组分支，PyROOT会自动为这类分支生成可直接访问的属性。但traj是结构化分支（包含多个不同类型的子成员），PyROOT无法自动推断其结构并生成对应的属性，所以需要我们手动映射或者用MakeClass自动生成映射类来处理。

内容的提问来源于stack exchange，提问作者Sadia Khalil