N*N 행렬의 제곱

# N*N행렬의 제곱을 구하는 함수
# matrix 에는 [[1,0],[0,1]] 과 같이 행렬이 N*N의 format으로 들어오면 된다.
# return 또한 input format 과 같이 return된다


def square(matrix):
    N = len(matrix)
    return_matrix = [[] for _ in range(N)]
    tmp = list(zip(*matrix))

    for i in range(N):
        for j in range(N):
            return_matrix[i].append(sum(map(lambda x : x[0] * x[1],zip(matrix[i], tmp[j]))))
    return return_matrix
'''
# testcase
print(square([[1,2,3],[1,2,3],[1,2,3]]))
'''