Question

如果我在python中有一个方形矩阵作为嵌套列表，我可以将它拆分成几个大小相等的框并创建一个新列表，其中每个元素都是其中一个框中的数字列表。 E.g。

a = [[1, 2, 3, 4],
     [5, 6, 7, 8],
     [9, 10, 11, 12],
     [13, 14, 15 ,16]]

b = [[a[0][0], a[0][1], a[1][0], a[1][1]],
     [a[0][2], a[0][3], a[1][2], a[1][3]],
     [a[2][0], a[2][1], a[3][0], a[3][1]],
     [a[2][2], a[2][3], a[3][2], a[3][3]]]

有更简单的方法吗？有没有办法将其设置为一个函数，我可以应用于不同大小的矩阵，并指定框的大小？

Answer 1

以下内容等同于您所拥有的内容，并且更简洁：

b = [a[0][:2] + a[1][:2],
     a[0][2:] + a[1][2:],
     a[2][:2] + a[3][:2],
     a[2][2:] + a[3][2:]]

或等同的列表理解：

b = [a[i][s] + a[i+1][s] for i in (0,2) for s in (slice(None,2), slice(2,None))]

Answer 2

使用NumPy：

In [31]: import numpy as np

In [32]: a = np.arange(1,17).reshape(4,4)

In [33]: a
Out[33]: 
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12],
       [13, 14, 15, 16]])

In [34]: b = a.reshape(-1,2,2,2).swapaxes(1,2).reshape(4,4)

In [35]: b
Out[35]: 
array([[ 1,  2,  5,  6],
       [ 3,  4,  7,  8],
       [ 9, 10, 13, 14],
       [11, 12, 15, 16]])

自适应函数来切割嵌套列表？

2 个答案: