我一直在寻找只得到非重叠子矩阵的方法。 下面的代码将查找所有子矩阵
代码:
n = 4
matrix = [[1,2,3,4],
[5,6,7,8],
[9,10,11,12],
[13,14,15,16]]
k = 2 #Finding 2x2 submatrices
t=[]
for i in range(n-k+1):
for j in range(n-k+1):
l=[]
for x in range(i,i+k):
for y in range(j,j+k):
if x==i or x==i+k-1 or y==j or y==j+k-1:
l.append(matrix[x][y])
t.append(l)
因此,如果我打印t
,我会得到:
for i in t:
print(i)
O/p:
[1, 2, 5, 6]
[2, 3, 6, 7]
[3, 4, 7, 8]
[5, 6, 9, 10]
[6, 7, 10, 11]
[7, 8, 11, 12]
[9, 10, 13, 14]
[10, 11, 14, 15]
[11, 12, 15, 16]
但我想要的o/p是:
[1,2,5,6]
[3,4,7,8]
[9,10,13,14]
[11,12,15,16]
当您在
range(n-k+1)
中的元素上循环时,您正在使用参数start=0、stop=n-k+1=3、step=1调用range
,这将导致数组[0, 1, 2]
。把它作为你的出发点是个问题要确保结果只包含互斥元素,请将步骤参数更改为k:
相关问题 更多 >
编程相关推荐