uproot问题:我在TBrowser中看到的完整TTree结构没有出现在uproot中

时间:2020-10-16 12:35:06

标签: python uproot

摘要:运行uproot tutorial时,TTree分支似乎丢失了。

我有一个根文件,其中包含一个名为“ prod”的TTree,它具有一组复杂的锯齿状的叶子和分支,可以在ROOT的TBrowser中看到。 我使用此根文件作为输入开始了uproot教程,并在交互式会话开始时收到以下错误:

>>>import uproot as up
>>> file = up.open('small.root')
>>> file
<ROOTDirectory b'small.root' at 0x025b5e3477c0>
>>> file.keys()
[b'prod;1']
>>> file.classnames()
Traceback (most recent call last):
  File "<pyshell#13>", line 1, in <module>
    file.classnames()
AttributeError: 'ROOTDirectory' object has no attribute 'classnames'
>>> file['prod']
<TTree b'prod' at 0x025b5dd15d00>

为什么在尝试获取类名时出现错误?

忽略此并继续。下一个问题是当我尝试查看TTree prod

中的内容时
>>> tree = file['prod']
>>> tree.keys()
[b'COSMIC', b'COSMICRES', b'COTNBC', b'COTTIME', b'GLB', b'LUM', b'MET', b'MU', b'PHOTON', b'RESIDUALS', b'TRACK', b'TRKDET', b'VERTEX', b'ZVTX', b'MOM_ntk', b'MOM_pt', b'MOM_px', b'MOM_py', b'MOM_pz']
>>> branches = tree.arrays(namedecode='utf-8')
>>> branches.keys()
dict_keys(['MET', 'MOM_ntk', 'MOM_pt', 'MOM_px', 'MOM_py', 'MOM_pz'])

uproot教程意味着我应该获得所有分支,但是显然我缺少很多分支。特别是,转移的唯一分支就是所谓的“简单”分支,因为它们只有数字数据成员。

其他分支中还包含其他项。例如,“ MU”分支具有事件中每个μon的属性。首先是此类muon的数量,然后是这些muon的每个属性的另一个分支,例如其质量和与之关联的轨道号。 “ MET”,“ MOM_ntk”,“ MOM_pt”,“ MOM_px”,“ MOM_py”和“ MOM_pz”都仅包含浮点数列表。每个事件MOM_ntk只有一个数字(称为“ alpha”),其他MOM分支中的每个事件都具有“ alpha”数字。

该文件只有1000个事件,总大小只有5MB。

我想知道其他所有分支都去了哪里!我的朋友在哪里“ COTNBC”或“ GLB”(其中应该包含所有的跑步和赛事编号)。

任何建议或帮助将不胜感激!


具有32GB RAM的Windows 10 PC 英特尔(R)酷睿TM i7-8700 CPU @ 3.20GHz x64处理器

1 个答案:

答案 0 :(得分:0)

根据您要执行的操作和所获得的输出,您似乎希望使用Uproot 4,但是您拥有Uproot3。直到Uproot 4与Uproot 3具有功能相同的功能,在名为uproot4(pip install uproot4import uproot4)的软件包中。这样一来,您就可以在过渡期间同时使用两者。

这些差异之一是Uproot 3的keys()方法没有遍历分支的子分支,而Uproot 4的方法则是通过分支的。 (这两个参数中都有一个recurse=True/False参数,但默认设置正是出于您所看到的原因而更改的。)

名称更改(uproot4uprootuprootuproot3)将在今年年底之前发生。 Uproot 4只需要写入文件即可。