1

I have two 2D JaggedArrays of the same length (in axis=0) that I'd like to stack in axis=1. For example:

a = awkward.fromiter([[0], [3, 4]])
b = awkward.fromiter([[1, 2], [5]])

and I want to get this JaggedArray:

[ [ [0],    [1, 2] ],
  [ [3, 4], [5]    ]  ]

I can get exactly this with the following:

awkward.fromiter(np.stack([a, b], axis=1))

but is there a way to accomplish this directly in awkward, without having to use fromiter() on nested numpy arrays?

mason
  • 11
  • 3

0 Answers0