Olá Tati,
Se você prestar atenção na composição de uma Matriz A com elementos [tex]a_{ij}[/tex], você irá perceber que quando j = i, isso irá representar a diangonal principal. Essa matriz que você está pedindo para construir chama-se a matriz identidade.
[tex]\left[\begin{array}{ccc}1&0&0\\0&1&0\\0&0&1\end{array}\right][/tex]
