https://www.somethingsblog.com/2024/10/18/mastering-numpys-trace-function-unlock-matrix-diagonals/