awkward-array - 如何在笨拙的数组中复制锯齿状数组

Question

在尴尬0中，我想分别保留表格的各种选择

在伪代码中

X = awkward.Table(...)

one_jet = X[X.n_jet == 1]
two_jet = X[X.n_jet == 1]

awkward.save(one_jet)
awkward.save(two_jet)

但我注意到任何索引锯齿状数组的内容都不会改变（只有开始停止变薄），因此天真地保存开始，以这种方式停止会导致磁盘上的数据重复。有没有办法“重新打包”锯齿状阵列，使它们再次变得密集？

Answer 1

我最近也遇到了这个问题。我找到的最干净的解决方案是从屏蔽数组构造一个新的笨拙数组，然后保存新数组。

X = awkward.Table(...)

one_jet = awkward.fromiter(X[X.n_jet == 1])
two_jet = awkward.fromiter(X[X.n_jet == 2]) # I assume you meant 2 here?

awkward.save(one_jet)
awkward.save(two_jet)