Demonstrate how to iterate diagonals in a matrix in python