查找3d列表

Question

我有一个列表l=[[[1,2,3],[4,5,6]],[[7,8,9],[9,1,2]]] 如何使用切片获取列表[[[1,2],[4,5]],[[7,8],[9,1]]]？我试过l[:][:][:2]，但它给了我整个3D列表。

Answer 1

你可以使用嵌套列表理解，当然这不仅仅是使用切片，但希望仍然有用：

>>> lst = [[[1,2,3],[4,5,6]],[[7,8,9],[9,1,2]]]
>>> [[subsub[:-1] for subsub in sublist] for sublist in lst]
[[[1, 2], [4, 5]], [[7, 8], [9, 1]]]