어떻게 두 개의 행렬을 입력으로 취할 수 있습니까?

Question

이 코드에서 행렬의 추가를 찾는 것으로 가정합니다. 나는이 프로그램을 실행하고 파이썬 쉘 A+B에 쓸 때 A+B

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

이 대답은, [[7, 4], [5로 나온다 0], [4, 4], [2, 2], [-3, 3], [-2, 4]]

대답은 실제로해야 파이썬 함수가 def addition (A,B) 입력으로 두 행렬을한다라고하고, 그래서 나는이 프로그램에 통합해야 할 무엇 [[9,6],[2,3],[2,8]]

결과로 두 입력의 추가와 함께 반환합니다.

Answer 1

아니면 중첩 된 지능형리스트 두려워하지 않으면 당신은 당신의 행렬에 대한 운영자 +를 오버로드하려는 경우, 당신은 당신은 2 차원을 포장해야 한 라이너

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

Answer 2

하여이 작업을 수행 할 수 있습니다 목록에 클래스로 넣고 def 메서드 __add__. 예를 들어 (귀하의 추가 기능을 사용합니다) :

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