Print Diagonal Elements Of Matrix In Python