创建嵌套列表并使用 for 循环内联将它们配对

Question

n 是我们的数组需要的 Input 总数。 因为目标是打印作为输入的一对数字。 我想找到下面提到的第 3 行的输出，当我在 python ide 中按 enter 时，我得到了这个结果。
我有点困惑，代码中的第三行是如何工作的。当我每天都在学习新事物时，这个问题就出现了，我被击中了第三行。

n = int(input())
arr = list(map(int, input().split()))
arr = [arr[i:i+2] for i in range(0, n*2, 2)]
print(arr)

第一个输入 -

4    
1 3 2 4 6 8 9 10   

第二个输入 -

4    
6 8 1 9 2 4 4 7

第一个输出 -

[[1, 3], [2, 4], [6, 8], [9, 10]]  

第二个输出 -

[[6, 8], [1, 9], [2, 4], [4, 7]]

Answer 1

像这样理解单行代码的一个好技巧是一步一步地扩展单行代码。

考虑一下这个输入：

n = 4
arr = [6, 8, 1, 9, 2, 4, 4, 7]

让我们从原来的第 3 行开始。

arr = [arr[i:i+2] for i in range(0, n*2, 2)]

首先，range(0, n*2, 2) 是range(0, 8, 2)，其功能类似于列表[0, 2, 4, 6]。所以现在我们正在处理这个：

arr = [arr[i:i+2] for i in [0, 2, 4, 6]]

注意arr[i:i+2] 指的是二元素数组[arr[i], arr[i+1]]。例如，当i=0，arr[i:i+2] = [arr[0], arr[1]] = [6, 8]。所以现在我们有：

arr = [[arr[i], arr[i+1]] for i in [0, 2, 4, 6]]

对每个 i 进行扩展：

arr = [[arr[0], arr[1]], [arr[2], arr[3]], [arr[4], arr[5]], [arr[6], arr[7]]]

将arr 的值代入得到最终结果：

[[6, 8], [1, 9], [2, 4], [4, 7]]

Answer 2

arr = [arr[i:i+2] for i in range(0, n*2, 2)] 等价于：

arr2 = []
for i in range(0, n*2, 2):
    arr2.append(arr[i:i+2])
arr = arr2

range(0, n*2, 2) 是一个迭代器，它产生从0 到（但不包括）n*2 的所有其他数字。

arr[i:i+2] 对arr 进行切片，将元素从ith 位置直到（但不包括）i+2th 位置。

所以，这个表达式有效地给你[arr[0,2], arr[2,4], arr[4,6], ..., arr[n*2-2,n*2]]。