我有两个相同长度的 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 数组上使用?