如何解决用uproot4读取树的一部分
我有一个带有单个ROOT树的文件,并希望在多个进程中使用uproot4读取它,每个进程都读取树的不连贯部分。
使用uproot3,可以通过传递entrysteps
的可迭代对象来实现:
def index_generator(rank,nranks,numentries,chunk_size):
start = int(numentries / nranks * rank)
stop = int(numentries / nranks * (rank + 1))
index = start
while index < stop:
yield (index,min(index + chunk_size,stop))
index += chunk_size
entrysteps = index_generator(rank,chunk_size)
iterator = uproot.tree.iterate(
filename,treename,branches=branches,namedecode="utf-8",entrysteps=entrysteps)
我的rootroot4 entrysteps
似乎已被step_size
取代,后者不再接受迭代器。
uproot4当前是否有一种方法?如果没有,是否计划这样的选择?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。