python-矩阵乘法

解题思路:

注意事项:

参考代码:

def f(A,n,m):  
    if m == 0:  
        C = [[1 if i == j else 0 for j in range(n)] for i in range(n)]  
    else:  
        C = A[:]  
        for i in range(m-1):  
            C = [[sum(a*b for a,b in zip(row,col)) for col in zip(*A)] for row in C]  
  
    for i in range(n):  
        for j in range(n):  
            print(C[i][j],end=' ')  
        print()  
      
      
if __name__ == '__main__':  
    n,m = map(int,input().strip().split())  
    A = [[int(j) for j in input().strip().split()] for i in range(n)]  
    f(A,n,m)