python - 如何将嵌套的整数列表拆分为列表

Question

lst = [[1, 5],  
       [2, 2]

这是我的嵌套列表，我需要列出这个点:

output = [[1, 5, 2, 2]

这是我的尝试，它适用于这种情况，但如果我有一个行长度为 6 或大于 4 的示例，则会失败

new_lst = []

for x in range(len(lst)):
    for y in range(0, len(lst[x]), 2):
        new_lst.append([lst[x][y],lst[x][y+1]])

counter_a = 0
counter_b = 1
output = []

while counter_b - 4 <= len(lst):
    output.append(new_lst[counter_a] + new_lst[counter_a + 2])
    output.append(new_lst[counter_b] + new_lst[counter_b + 2])
    counter_a += 4
    counter_b += 4

print(output)

Answer 1

这个怎么样？这对于所有大小为 nxm 的列表都是通用的，其中 n 和 m 是偶数。逻辑是在行和列中以步长 2 进行迭代，然后取出 2x2 元素 block 并将其附加到输出列表。

lst = [[1, 6, 5, 6],  
       [2, 5, 6, 8],  
       [7, 2, 8, 1],  
       [4, 4, 7, 3]]
output = []

for j in range(0, len(lst), 2):
     for i in range(0, len(lst[0]), 2):
         output.append([lst[j][i], lst[j][i+1], lst[j+1][i], lst[j+1][i+1]])

output : [[1, 6, 2, 5], [5, 6, 6, 8], [7, 2, 4, 4], [8, 1, 7, 3]]