Pergunta de entrevista da empresa Capital One

Solve a matrix in Python