PyROOT读取ROOT文件异常:分支被识别为叶子无法访问子叶求助
解决PyROOT提取自定义类分支数据的问题
问题分析
你遇到的核心问题是PyROOT未加载自定义类MCInfo的字典,导致无法识别分支对应的对象类型,进而出现mcinfo为空、无法访问对象成员的情况。另外,MC是自定义对象分支,而非基础类型分支,所以GetListOfLeaves()只会显示分支本身,其成员变量属于对象属性,并非TLeaf,这是正常现象。
解决方案
1. 先加载自定义类的字典
ROOT的自定义类(如MCInfo)需要先让PyROOT识别其结构,才能正确解析分支数据。有两种方式加载:
- 如果有编译好的共享库(比如
libMCInfo.so):
import ROOT ROOT.gSystem.Load("libMCInfo.so")
- 如果只有头文件(
MCInfo.h):
import ROOT ROOT.gInterpreter.Declare('#include "MCInfo.h"')
2. 正确绑定分支地址(两种写法)
写法一:手动设置分支地址
infile = ROOT.TFile.Open(fname) data = infile.Get("data") # 实例化MCInfo对象 mcinfo = ROOT.MCInfo() # 绑定分支到对象 data.SetBranchAddress("MC", mcinfo) # 遍历事件提取数据 for entry_idx in range(data.GetEntries()): data.GetEntry(entry_idx) # 直接访问对象成员,比如energy print(mcinfo.energy)
写法二:PyROOT自动分支绑定(更简洁)
PyROOT支持直接遍历TTree的条目,并通过属性访问分支对象:
infile = ROOT.TFile.Open(fname) data = infile.Get("data") # 可选:关闭无关分支提升速度 data.SetBranchStatus("*", 0) data.SetBranchStatus("MC", 1) # 直接遍历条目,访问MC分支的成员 for entry in data: print(entry.MC.energy)
为什么原代码失效?
C++代码中file->GetList()->FindObject("MC")是旧写法,用于获取类的TClass对象,但PyROOT不需要这种操作——只要加载了类字典,直接实例化ROOT.MCInfo()即可。原代码未加载类字典,导致无法识别MCInfo类型,自然获取不到有效对象。
内容的提问来源于stack exchange,提问作者Alheana
相关产品推荐
相关产品推荐

