首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >uproot3可以写入包含数组的TBranches吗?

uproot3可以写入包含数组的TBranches吗?
EN

Stack Overflow用户
提问于 2020-12-19 22:17:10
回答 1查看 85关注 0票数 2

在uproot 3文档中有这样信息: uproot只能写入每个条目包含1个值的分支。另一方面,我可以在uproot上看到一些关于编写锯齿数组等的主题。所以,我想确保: uproot可以将包含数组的TBranches写入TTree吗?如果是这样的话,有记录在案吗?

谢谢!

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2020-12-19 23:00:07

当它被移植到uroot4时,这将被更好地记录下来,但是我们现在拥有的关于在uroot3中编写锯齿数组的最好的文档是pull请求和相关的问题(所有这些都是相互关联的):

https://github.com/scikit-hep/uproot3/pull/477

以下是测试中的一个示例:

代码语言:javascript
复制
import uproot3
import awkward0

a = awkward0.fromiter([[0],
                       [1, 2],
                       [10, 11, 12]])

with uproot3.recreate(filename, compression=None) as f:
    f["t"] = uproot3.newtree({
        "branch": uproot3.newbranch(numpy.dtype(">i4"), size="n")
    })
    f["t"].extend({"branch": a, "n": [1, 2, 3]})

f = ROOT.TFile.Open(filename)
tree = f.Get("t")
for i, event in enumerate(tree):
    assert(numpy.all([x for x in event.branch] == a[i]))
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/65370728

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档