uproot 中数组的零填充可变长度

Question

我需要一种内存有效的方法来根据树中的最大数组对每个事件的不均匀数组大小进行零填充。有什么办法可以在 uproot 中做到这一点？

Answer 1

jagged_array.pad(max_length).fillna(0)

如果需要求最大值，可以

jagged_array.counts.max()

如果您随后需要将其转换为纯 NumPy，请使用 regular()。