如何将两个矩阵作为输入

Question

In this code I'm suppose to find the addition of the matrices. A+B

[[x + y for x,y in zip(w,v)] for w,v in zip(A,B)]

When I run the program and write in the python shell A+B The answer comes out as [[7, 4], [5, 0], [4, 4], [2, 2], [-3, 3], [-2, 4]]

The answer should actually be [[9,6],[2,3],[2,8]]

What do I need to integrate in the program so the Python function called def addition (A,B) takes the two matrices as input and return with the addition of the two input as the result.

Answer 1

或者，如果您不害怕嵌套列表推导，则可以单线执行

C = [[x + y for x,y in zip(w,v)] for w,v in zip(A,B)]

Answer 2

If you want to overload the operator + for your matrices you have to wrap the 2 dimensional list into a class and def the method __add__ . For example (I use your addition function):

>>> class Matrix(object):
    @staticmethod
    def addition (A, B):
        d=[]
        n=0
        while n < len(B):
            c = []
            k = 0
            while k < len (A[0]):
                c.append(A[n][k]+B[n][k])
                k=k+1
            n+=1
            d.append(c)
        return d
    def __init__(self,lst):
        self.lst=lst
    def __add__(self, other):
        return Matrix(Matrix.addition(self.lst, other.lst))
    def __repr__(self):
        return str(self.lst)


>>> A=Matrix([[7,4], [5,0], [4,4]])
>>> B=Matrix([[2,2], [-3,3], [-2, 4]])
>>> A+B
[[9, 6], [2, 3], [2, 8]]