在第二维上将列表扩展相同的索引

时间:2019-03-18 15:44:30

标签: python list indexing extend

我想了解如何通过相同的索引扩展列表。如 A是形状(2,3,2)。 B是形状(2,5,2)。我想得到形状为(2,8,2)的C。 谢谢。

A=[[[1, 2], [2, 3], [3, 4]], [[6, 7], [7, 8], [8, 9]]]
B=[[[11, 12], [12, 13]], [[16, 17], [17, 18]]]

我想得到

[[[1, 2], [2, 3], [3, 4],[11, 12], [12, 13]],
 [[6, 7], [7, 8], [8, 9], [16, 17], [17, 18]]]

1 个答案:

答案 0 :(得分:0)

我通过使用np.hstack((a,b))解决了这个问题。其中a和b是A和B的np.array。 然后将其转换回列表。感谢您的宝贵时间。