如何使用python在矩阵中找到最大平方和

Question

我正在尝试编写一个代码，该代码将为我提供矩阵中最大平方和。

注意：在形成正方形时，这些值必须彼此相邻。

注意 2：我只被允许使用函数，目前我正在上课的课程中不允许导入任何东西。

例如：

 max_mat_square([[1,2.5,3],

                 [4,-4,6],

                 [0.5,-1,5]])

[7.5]

现在这是我的代码：

def max_mat_square(mat):

    sum1 = []

    for i in range(len(mat[:-1])):

        print i

        for j in range(len(mat[1:])):

                print 'index' , j

                sum1.append([float(mat[i][j]) + float(mat[i+1][j])+float(mat[i][j+1])+float(mat[i+1][j+1])])

    return max(sum1)

现在代码在第一个例子中工作正常，但是每当我运行这个矩阵时抛出它：

max_mat_square([[1,2], [3,4], [5,6]])

我收到“列表索引超出范围”错误，我想要这个矩阵的结果是 18。

我尝试了很多方法来修复错误，但都没有奏效。

Answer 1

我认为你的第一个例子只是偶然地起作用，因为你的矩阵的形状。目前，该循环中使用的切片意味着您正在从第二个子列表中选择到最后：mat[1:]>> [[3,4], [5,6]]这意味着您将尝试遍历每个range(2)将超出范围的元素。将第二个 for 语句更改如下：def max_mat_square(mat):    sum1 = []    for i in range(len(mat[:-1])):        for j in range(len(mat[i])-1):            print (i , j)            sum1.append([float(mat[i][j]) + float(mat[i+1][j])+float(mat[i][j+1])+float(mat[i+1][j+1])])    return max(sum1)

Answer 2

问题是你的内for循环。您应该j从 0 开始到该行的倒数第二个索引：for j in range(len(mat[i][:-1])):    ...所以你的最终功能变成：def max_mat_square(mat):    sum1 = []    for i in range(len(mat[:-1])):        print(i)        for j in range(len(mat[i][:-1])):                print('index' , j)                sum1.append([float(mat[i][j]) + float(mat[i+1][j])+float(mat[i][j+1])+float(mat[i+1][j+1])])    return max(sum1)