1

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

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]]]

有没有更简单的方法来做到这一点？有没有办法将其设置为可以应用于不同大小的矩阵并指定框的大小的函数？

4

2 回答 2

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))]
于 2013-02-15T18:44:32.553 回答
1

使用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]])
于 2013-02-15T18:51:29.817 回答