python - 如何遍历嵌套列表的一部分？

Question

我有这个嵌套列表:

nested = [[1, 1, 1], [2, 2, 2], [3, 3, 3]]

我想从内部列表的前 2 个元素中减去 1，所以我尝试了列表理解:

nested = [[x - 1 for x in stack[0:2]] for stack in nested]

它确实返回了内部列表的前 2 个元素减 1，但它完全删除了最后一个元素

nested = [[0, 0], [1, 1], [2, 2]]

我以为通过切片列表，它不会影响其他元素。但是在这种情况下它不起作用。谁能给我解释一下？

Answer 1

要保留第 3 个元素，请将其包含在列表理解中:

>>> [ [x - 1 for x in stack[0:2]] + stack[2:] for stack in nested ]
[[0, 0, 1], [1, 1, 2], [2, 2, 3]]

以上适用于任何长度的 stack。

或者，如果堆栈总是恰好包含三个元素:

>>> [[x-1, y-1, z] for x, y, z in nested]
[[0, 0, 1], [1, 1, 2], [2, 2, 3]]

或者，您可以就地进行更改:

>>> for stack in nested: stack[0]-=1; stack[1]-=1
... 
>>> nested
[[0, 0, 1], [1, 1, 2], [2, 2, 3]]