如何在不使用numpy的情况下获得所有的对角二维表？

def get_all_diagonal_tables(matrix):
    rows = len(matrix)
    cols = len(matrix[0])
    tables = []

    # 获取主对角线
    main_diagonal = []
    for i in range(min(rows, cols)):
        main_diagonal.append(matrix[i][i])
    tables.append(main_diagonal)

    # 获取副对角线
    if rows >= cols:
        for i in range(cols):
            diagonal = []
            r = 0
            c = i
            while r < rows and c >= 0:
                diagonal.append(matrix[r][c])
                r += 1
                c -= 1
            tables.append(diagonal)
    else:
        for i in range(rows):
            diagonal = []
            r = i
            c = cols - 1
            while r < rows and c >= 0:
                diagonal.append(matrix[r][c])
                r += 1
                c -= 1
            tables.append(diagonal)

    return tables

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

diagonal_tables = get_all_diagonal_tables(matrix)
for table in diagonal_tables:
    print(table)

[1, 5, 9]
[2, 6]
[3]
[4, 8]
[7]