python矩阵加法例题,Python中的矩阵加法-列表

I'm trying to write Matrix Addition function using Python.

Here is the code I've been trying, but it gives me a list index error and I cannot figure out why.

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

using the following lists:

A = [[2,4], [7,0], [6,3]]

B = [[3,1], [-1,8], [-3, 3]]

So in theory, A[0][0] + B[0][0] would equal 5, and I would want to add that value to position Z[0][0].

However I keep receiving the error: IndexError: list index out of range

解决方案

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

As for how you could fix your current code:

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)

The important parts here being that you cannot just assign to Z[i][j] unless that row/column already exists, so you need to construct each inner list separately and append them to Z. Also the inner loop needs to end at the length of a row, so I changed range(len(A)) to range(len(A[i])).