suma resta matriz

def suma_resta_matrices(A,B,operacion):
    C=[0 for i in range(len(A))]
    for i in range(len(C)):
        C[i]=[0 for i in range(len(A[0]))]
    if operacion=="suma":
        for i in range(len(A)):
            if len(A)==len(A[i]) and len(B)==len(B[i]) and len(A)==len(B) and len(A[i])==len(B[i]):
                for j in range(len(A[i])):
                    C[i][j]=A[i][j]+B[i][j]
            else:
                return "Error"
    elif operacion=="resta":
        for i in range(len(A)):
            if len(A)==len(A[i]) and len(B)==len(B[i]) and len(A)==len(B) and len(A[i])==len(B[i]):
                for j in range(len(A[i])):
                    C[i][j]=A[i][j]-B[i][j]
            else:
                return "Error"

    for i in range(len(A)): #Esto es para que me de la matriz resultado en forma de matriz
        a=""
        for j in range(len(A[0])):
            a=a+str(C[i][j])+" "
        print(a)

    return C #También dejo este return C para que también me lo devuelva como lista

#El programa tiene muchas condiciones, si A y B no son del mismo tamaño o hay alguna fila o columna que no concuerda con las demás, da "Error"

print(suma_resta_matrices([[1,2,3],[3,2,1],[2,1,3]],[[4,5,6],[5,6,4],[6,4,5]],"resta"))