我需要对列表列表进行切片:
A = [[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]] idx = slice(0,4) B = A[:][idx]
上面的代码没有给我正确的输出。
我想要的是:[[1,2,3],[1,2,3],[1,2,3]]
[[1,2,3],[1,2,3],[1,2,3]]
使用切片对象很少比使用列表理解更容易阅读,这不是其中之一。
>>> A = [[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]] >>> [sublist[:3] for sublist in A] [[1, 2, 3], [1, 2, 3], [1, 2, 3]]
这是非常清楚的。对于每个子列表A,给我前三个元素的列表。
A