Question

我正在尝试使用Python编写Matrix Addition函数。

这是我一直在尝试的代码，但它给了我一个列表索引错误，我无法弄清楚原因。

def matrixADD(A,B):
Z = []
#TODO
for i in range(0,len(A)):
    for column in range(0, len(A)):
        result = A[i][column] + B[i][column]
        Z[i][column] = (result)
return Z

使用以下列表：

A = [[2,4], [7,0], [6,3]]
B = [[3,1], [-1,8], [-3, 3]]

所以从理论上讲，A [0] [0] + B [0] [0]等于5，我想将该值加到Z [0] [0]位置。

但是我一直收到错误：IndexError：列表索引超出范围

Answer 1

>>> A = [[2,4], [7,0], [6,3]]
>>> B = [[3,1], [-1,8], [-3, 3]]
>>> Z = [map(sum, zip(*t)) for t in zip(A, B)]
>>> Z
[[5, 5], [6, 8], [3, 6]]

至于如何修复当前代码：

Z = []
for i in range(len(A)):
    row = []
    for j in range(len(A[i])):
        row.append(A[i][j] + B[i][j])
    Z.append(row)

这里的重要部分是您不能只分配给Z[i][j]，除非该行/列已经存在，因此您需要单独构建每个内部列表并将它们附加到Z。内部循环也需要以行的长度结束，因此我将range(len(A))更改为range(len(A[i]))。

Answer 2

len(A) = 3但是矩阵的维度为3x2，因此当您尝试访问A [2] [2]时（因为列在0和len（A）之间），您将超出界限。

Answer 3

对于column，您使用范围0到len（A）（即3）。 A [i] [2]将超出范围，因为A [i]的长度仅为2.

尝试使用列范围结束len（A [i]）而不是len（A）：

def matrixADD(A,B):
    Z = []
    #TODO
    for i in range(0,len(A)):
        for column in range(0, len(A[i])):
            result = A[i][column] + B[i][column]
            Z[i][j] = (result)
    return Z

Python中的Matrix加法 - 列表

3 个答案: