diags.py

from pprint import pprint

def diags(matrix):
    x = len(matrix[0])
    
    yield [matrix[i][i] for i in range(x)]
    yield [matrix[i][x - 1 - i] for i in range(x)]
    for i in range(1, x):
        row1 = []
        row2 = []
        row3 = []
        row4 = []
        for j in range(i):
            row1.append(matrix[x - i + j][j])
            row2.append(matrix[j][x - i + j])

            row3.append(matrix[j][i-1 - j])
            row4.append(matrix[x - 1 - j][x - i + j])
        yield row1
        yield row2
        yield row3
        yield row4
      
        

matr = [[i * j for j in range(15)] for i in range(15)]

pprint(matr)
print()

for row in diags(matr):
    print(row)