1

我有两个相同长度的 2D JaggedArrays(轴 = 0),我想在轴 = 1 中堆叠。例如:

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

我想得到这个 JaggedArray:

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

我可以通过以下方式得到这个:

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

但是有没有办法直接在中完成此操作awkward,而不必fromiter()在嵌套的 numpy 数组上使用?

4

0 回答 0