matrix.py

##
## EPITECH PROJECT, 2018
## 108trigo_2017
## File description:
## Matrix file
##

from math import *

def identity_mat(n):
    tmp = []
    for i in range(n):
        ident = []
        for j in range(n):
            ident.append(1 if j == i else 0)
        tmp.append(ident)
    return tmp

def init_mat(n, k):
    tmp = []
    for i in range(n):
        ident = []
        for j in range(n):
            ident.append(k)
        tmp.append(ident)
    return tmp

def mat_mult(mat1, mat2):
    tmp = []
    for i in range(len(mat1)):
        mult = []
        for j in range(len(mat2[0])):
            a = 0
            for k in range(len(mat1[0])):
                a += mat1[i][k] * mat2[k][j]
            mult.append(a)
        tmp.append(mult)
    return tmp

def pow_mat(mat, n):
    tmp = mat
    for i in range(n - 1):
        tmp = mat_mult(tmp, mat)
    return tmp

def div_mat(mat, k):
    for i in range(len(mat)):
        for j in range(len(mat[0])):
            mat[i][j] /= k
    return mat

def add_mat(mat1, mat2):
    tmp = []
    for i in range(len(mat1)):
        c = []
        for j in range(len(mat1[0])):
            c.append(mat1[i][j] + mat2[i][j])
        tmp.append(c)
    return tmp

def sub_mat(mat1, mat2):
    tmp = []
    for i in range(len(mat1)):
        c = []
        for j in range(len(mat1[0])):
            c.append(mat1[i][j] - mat2[i][j])
        tmp.append(c)
    return tmp